Member-only story
Machine Learning
A Beginners Guide To Five Useful Pytorch Tensor Tools
An overview of useful tensor operations that help in matrix manipulation, conditional operations, and reshaping techniques
3 min readJan 25, 2021
An overview of a few useful tensor operations that help in matrix manipulation, conditional operations, and reshaping techniques.
This article is a summation of a notebook completed as a part of the 6-week free certification course, “Deep Learning with PyTorch: Zero to GANs” taught by Aakash N S and offered by freecodecamp.org.
PyTorch is an open-source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab. This article describes five very useful tensor operations.
- torch. Reshape
- torch. Narrow
- torch.Numel
- torch. Mean
- torch. Flatten
Before we begin, let’s install and import PyTorch.
