We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
8935dbc
TorchShard: Slicing a PyTorch Tensor Into Parallel Shards. It has the same API design as PyTorch.