API Reference
Casts all floating point parameters and buffers to bfloat16 datatype.
bfloat16
Note
This method modifies the module in-place.
Module: self
previous
MnistModel.apply
next
MnistModel.buffers