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 tensorReturn 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.]]])
Article Tags :