Skip to main content
Log in

Mojo module

tensor_dict

Structs used for maintaining a collection of tensors.

Structs

  • TensorDict: A collection of keyed Tensor values used with checkpoint files.