Python - PyTorch atan2() method

Python Pytorch eye() method

Last Updated : 22 Apr, 2020

PyTorch is an open-source machine learning library developed by Facebook. It is used for deep neural network and natural language processing purposes. The function torch.eye() returns a returns a 2-D tensor of size n*m with ones on the diagonal and zeros elsewhere.

Syntax: torch.eye(n, m, out=None) Parameters: n: the number of rows m: the number of columns. Default - n out (Tensor, optional): the output tensor Return type: A 2-D tensor

Code #1:

Python3

# Importing the PyTorch library
import torch


# Applying the eye function and
# storing the resulting tensor in 'a'
a = torch.eye(3, 4)
print("a = ", a)

b = torch.eye(3, 3)
print("b = ", b)

c = torch.eye(5, 1)
print("c = ", c)

Output:

a =  tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])
b =  tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])
c =  tensor([[1.],
        [0.],
        [0.],
        [0.],
        [0.]])

Python - PyTorch atan2() method

S

sanskar27jain

Improve

Article Tags :

Practice Tags :

python

Similar Reads

Python PyTorch log2() method

PyTorch log2() method computes the logarithm to the base 2 of the elements of an input tensor. Â It computes the logarithm values element-wise. Â It takes a tensor as an input and returns a new tensor with computed logarithm values. The elements of the input tensor must be between zero and the positiv

Python - PyTorch frac() method

PyTorch torch.frac() method computes the fractional portion of each element in input. Syntax: torch.frac(input, out=None) Arguments input: This is input tensor. out: The output tensor. Return: It returns a Tensor. Let's see this concept with the help of few examples: Example 1: Python3 # Importing t

Python - PyTorch atan2() method

PyTorch atan2() method computes element-wise arctangent of (y/x), where y, x are the tensors with y-coordinates and x-coordinates of the points respectively. The arctangent of (y/x) is the angle between the positive x-axis and the line from (0,0) to the (x,y). So the atan2() method computes these an

Python | sympy.eye() method

With the help of sympy.eye() method, we can find the identity matrix by using sympy.eye() method. Syntax : sympy.eye() Return : Return an identity matrix. Example #1 : In this example, we can see that by using sympy.eye() method, we are able to find identity matrix having dimension nxn, where n will

Python - PyTorch numel() method

PyTorch torch.numel() method returns the total number of elements in the input tensor. Syntax: torch.numel(input) Arguments input: This is input tensor. Return: It returns the length of the input tensor. Let's see this concept with the help of few examples: Example 1: Python3 # Importing the PyTorch

Python - PyTorch is_tensor() method

PyTorch torch.is_tensor() method returns True if the passed object is a PyTorch tensor. Syntax: torch.is_tensor(object) Arguments object: This is input tensor to be tested. Return: It returns either True or False. Let's see this concept with the help of few examples: Example 1: Python3 # Importing t

Python - PyTorch is_storage() method

PyTorch torch.is_storage() method returns True if obj is a PyTorch storage object. Syntax: torch.is_storage(object) Arguments object: This is input tensor to be tested. Return: It returns either True or False. Let's see this concept with the help of few examples: Example 1: Python3 # Importing the P

Python - PyTorch is_storage() method

PyTorch torch.is_storage() method returns True if obj is a PyTorch storage object. Syntax: torch.is_storage(object) Arguments object: This is input tensor to be tested. Return: It returns either True or False. Let's see this concept with the help of few examples: Example 1: Python3 # Importing the P

Python PyTorch from_numpy()

PyTorch is an open-source machine learning library developed by Facebook. It is used for deep neural network and natural language processing purposes. The function torch.from_numpy() provides support for the conversion of a numpy array into a tensor in PyTorch. It expects the input as a numpy array

RandomResizedCrop() Method in Python PyTorch

In this article, we are going to discuss RandomResizedCrop() method in Pytorch using Python. RandomResizedCrop() method RandomResizedCrop() method of torchvision.transforms module is used to crop a random area of the image and resized this image to the given size. This method accepts both PIL Image