Open In App
Related Articles

Python PyTorch zeros() method

Improve
Improve
Improve
Like Article
Like
Save Article
Save
Report issue
Report
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.zeros() returns a tensor filled with the scalar value 0, with the shape defined by the variable argument size.
Syntax: torch.zeros(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 0, of same shape as size.
Code #1:
# Importing the PyTorch library
import torch
  
  
# Applying the zeros function and
# storing the resulting tensor in 't'
a = torch.zeros([3, 4])
print("a = ", a)
  
b = torch.zeros([1, 5])
print("b = ", b)
  
c = torch.zeros([5, 1])
print("c = ", c)
  
d = torch.zeros([3, 3, 2])
print("d = ", d)

                    
Output:
a =  tensor([[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.]])
b =  tensor([[0., 0., 0., 0., 0.]])
c =  tensor([[0.],
        [0.],
        [0.],
        [0.],
        [0.]])
d =  tensor([[[0., 0.],
         [0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]]])

 

Last Updated : 22 Apr, 2020
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads