MNistNet.bfloat16

MNistNet.bfloat16() → T

Casts all floating point parameters and buffers to bfloat16 datatype.

Note

This method modifies the module in-place.

Returns:

Module: self

previous

MNistNet.apply

next

MNistNet.buffers