Python Pytorch ones() 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.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:
import torch
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.]]])
Last Updated :
22 Apr, 2020
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...