SelfAttn2D.half

SelfAttn2D.half() T[source]

Casts all floating point parameters and buffers to half datatype.

Note

This method modifies the module in-place.

Returns:

Module: self