MNistNet.type

MNistNet.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

previous

MNistNet.train

next

MNistNet.xpu