Python Pytorch full() method
Last Updated :
22 Apr, 2020
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.full()
returns a tensor of size size filled with fill_value.
Syntax: torch.ones(size, fill_value, out=None)
Parameters:
size: a sequence of integers defining the shape of the output tensor
fill_value: the number to fill the output tensor with.
out (Tensor, optional): the output tensor
Return type: A tensor
Code #1:
import torch
a = torch.full([ 3 , 4 ], 3 )
print ( "a = " , a)
b = torch.full([ 2 , 5 ], 3.5 )
print ( "b = " , b)
|
Output:
a = tensor([[3., 3., 3., 3.],
[3., 3., 3., 3.],
[3., 3., 3., 3.]])
b = tensor([[3.5000, 3.5000, 3.5000, 3.5000, 3.5000],
[3.5000, 3.5000, 3.5000, 3.5000, 3.5000]])
Share your thoughts in the comments
Please Login to comment...