Python PyTorch zeros() 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.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:
import torch
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
Share your thoughts in the comments
Please Login to comment...