MnistModel.bfloat16#

MnistModel.bfloat16() T#

Casts all floating point parameters and buffers to bfloat16 datatype.

Note

This method modifies the module in-place.

Returns:

Module: self