torch.unbind — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch >
• torch.unbind
• 

Shortcuts

torch.unbind(input, dim=0) → seq¶

Removes a tensor dimension.

Returns a tuple of all slices along a given dimension, already without it.

Parameters

• input (Tensor) – the tensor to unbind
• dim (int) – dimension to remove

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]))

• torch.unbind
  • unbind()