Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
4 years ago

5 PyTorch Functions That You Will Definitely Need

Introduction To PyTorch

PyTorch is a free and open-source machine learning library developed by Facebook's AI Research lab. It can used for a range of applications, from Computer Vision to Natural Language Processing. It is an important and widely accepted tool for those designing machine learning projects.

Below are 5 functions of PyTorch that will be useful to any beginner:

  • as_tensor()
  • expand()
  • matmul()
  • round()
  • rsqrt()
# Import torch and other required modules
import torch
import numpy as np

Function 1 - torch.as_tensor()

tensor() is used to create a tensor type object for various computations. as_tensor() is the call by reference version of tensor()

Example 1 - working