torch.unbind
-
torch.unbind(input, dim=0) → seq
-
Removes a tensor dimension.
Returns a tuple of all slices along a given dimension, already without it.
Example:
>>> torch.unbind(torch.tensor([[1, 2, 3], >>> [4, 5, 6], >>> [7, 8, 9]])) (tensor([1, 2, 3]), tensor([4, 5, 6]), tensor([7, 8, 9]))
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.unbind.html