SelfAttn3D.type

SelfAttn3D.type(dst_type: dtype | str) → T

Casts all parameters and buffers to dst_type.

Note

This method modifies the module in-place.

Args:

dst_type (type or string): the desired type

Returns:

Module: self