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

previous

MnistModel.apply

next

MnistModel.buffers