Python Pytorch ones() method

0 0 1 minute read

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.ones() returns a tensor filled with the scalar value 1, with the shape defined by the variable argument size.

Syntax: torch.ones(size, out=None)

Parameters:
size: a sequence of integers defining the shape of the output tensor
out (Tensor, optional): the output tensor

Return type: A tensor filled with scalar value 1, of same shape as size.

Code #1:

# Importing the PyTorch library 
import torch 
  
  
# Applying the ones function and 
# storing the resulting tensor in 't' 
a = torch.ones([3, 4]) 
print("a = ", a) 
  
b = torch.ones([1, 5]) 
print("b = ", b) 
  
c = torch.ones([5, 1]) 
print("c = ", c) 
  
d = torch.ones([3, 3, 2]) 
print("d = ", d) 

Output:

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

        [[1., 1.],
         [1., 1.],
         [1., 1.]],

        [[1., 1.],
         [1., 1.],
         [1., 1.]]])