Método Python Pytorch ones()

PyTorch es una biblioteca de aprendizaje automático de código abierto desarrollada por Facebook. Se utiliza para fines de procesamiento de lenguaje natural y redes neuronales profundas.

La función torch.ones()devuelve un tensor relleno con el valor escalar 1, con la forma definida por el tamaño del argumento variable.

Sintaxis : torch.ones (tamaño, fuera = Ninguno)

Parámetros :
tamaño : una secuencia de números enteros que definen la forma del tensor de
salida (Tensor, opcional) : el tensor de salida

Tipo de valor devuelto: un tensor relleno con un valor escalar 1, de la misma forma que el tamaño .

Código #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)

Producción:

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.]]])