module
tensor_dict
Structs used for maintaining a collection of tensors.
Structs
-
TensorDict
: A collection of keyedTensor
values used with checkpoint files.
module
Structs used for maintaining a collection of tensors.
TensorDict
: A collection of keyed Tensor
values used with checkpoint files.