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 tensorReturn type: A tensor filled with scalar value 1, of same shape as size.
Code #1:
# Importing the PyTorch library import torch # Applying the ones function and # storing the resulting tensor in 't' 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.]]])
