Python Pytorch empty() 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:

# 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 empty() method

S

sanskar27jain

Improve

Article Tags :

Practice Tags :

python

Similar Reads

Python Pytorch empty() method

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.empty() returns a tensor filled with uninitialized data. The shape of the tensor is defined by the variable argument size. Syntax:

Python - PyTorch exp() method

PyTorch torch.exp() method returns a new tensor after getting the exponent of the elements of the input tensor. Syntax: torch.exp(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

Python - PyTorch ceil() method

PyTorch torch.ceil() method returns a new tensor having the ceil value of the elements of input, Which is the smallest integer larger than or equal to each element. Syntax: torch.ceil(inp, out=None) Arguments inp: This is input tensor. out: The output tensor. Return: It returns a Tensor. Let's see t

Python - PyTorch add() method

PyTorch torch.add() method adds a constant value to each element of the input tensor and returns a new modified tensor. Syntax: torch.add(inp, c, out=None) Arguments inp: This is input tensor. c: The value that is to be added to every element of tensor. out: This is optional parameter and it is the

Python - PyTorch div() method

PyTorch torch.div() method divides every element of the input with a constant and returns a new modified tensor. Syntax: torch.div(inp, other, out=None) Arguments inp: This is input tensor. other: This is a number to be divided to each element of input inp. out: The output tensor. Return: It returns

Python - PyTorch log() method

PyTorch torch.log() method gives a new tensor having the natural logarithm of the elements of input tensor. Syntax: torch.log(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:

Python | PyTorch asin() method

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.asin() provides support for the inverse sine function in PyTorch. It expects the input to be in the range [-1, 1] and gives the out

Python - PyTorch fmod() method

PyTorch torch.fmod() method gives the element-wise remainder of division by divisor. The divisor may be a number or a Tensor. Syntax: torch.fmod(input, div, out=None) Arguments input: This is input tensor. div: This may be a number or a tensor. out: The output tensor. Return: It returns a Tensor. Le

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: # Importing the PyTor

Python | PyTorch cosh() method

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.cosh() provides support for the hyperbolic cosine function in PyTorch. It expects the input in radian form. The input type is tens