PyTorch .unbind()

Anonymous contributor's avatar
Anonymous contributor
Published Jan 25, 2025
Contribute to Docs

In PyTorch, the .unbind() function removes the specified dimension and returns a tuple of tensors, each representing a slice along the given dimension.

  • Career path

    Data Scientist: Machine Learning Specialist

    Machine Learning Data Scientists solve problems at scale, make predictions, find patterns, and more! They use Python, SQL, and algorithms.
    • Includes 27 Courses
    • With Professional Certification
    • Beginner Friendly.
      95 hours
  • Free course

    Intro to PyTorch and Neural Networks

    Learn how to use PyTorch to build, train, and test artificial neural networks in this course.
    • Intermediate.
      3 hours

Syntax

torch.unbind(tensor, dim=0)
  • tensor: The tensor to unbind.
  • dim: The dimension to remove from the tensor. If not specified, the default value (0) is used.

Example

The following example demonstrates the usage of the .unbind() function:

import torch


# Create a tensor
ten = torch.tensor([[1, 2, 3],
                    [4, 5, 6],
                    [7, 8, 9]])


# Unbind the tensor along the first dimension (dim=0)
unbound = torch.unbind(ten)


# Print the resulting tuple
print(unbound)

The code above should output the following:

(tensor([1, 2, 3]), tensor([4, 5, 6]), tensor([7, 8, 9]))

All contributors

Contribute to Docs

Learn PyTorch on Codecademy

  • Career path

    Data Scientist: Machine Learning Specialist

    Machine Learning Data Scientists solve problems at scale, make predictions, find patterns, and more! They use Python, SQL, and algorithms.
    • Includes 27 Courses
    • With Professional Certification
    • Beginner Friendly.
      95 hours
  • Free course

    Intro to PyTorch and Neural Networks

    Learn how to use PyTorch to build, train, and test artificial neural networks in this course.
    • Intermediate.
      3 hours