MnistModel.type#

MnistModel.type(dst_type: Union[dtype, str]) T#

Casts all parameters and buffers to dst_type.

Note

This method modifies the module in-place.

Args:

dst_type (type or string): the desired type

Returns:

Module: self