SelfAttn3D.bfloat16
- SelfAttn3D.bfloat16() T[source]
Casts all floating point parameters and buffers to
bfloat16datatype.Note
This method modifies the module in-place.
- Returns:
Module: self
Casts all floating point parameters and buffers to bfloat16 datatype.
Note
This method modifies the module in-place.
Module: self