Método Python Pytorch full()

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.full()devuelve un tensor de tamaño size lleno de fill_value.

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

Parámetros :
tamaño : una secuencia de enteros que definen la forma del tensor de salida
valor_relleno : el número con el que rellenar el tensor de salida.
out (Tensor, opcional) : el tensor de salida

Tipo de retorno : un tensor

Código #1:

# Importing the PyTorch library
import torch
  
  
# Applying the full function and
# storing the resulting tensor in 'a'
a = torch.full([3, 4], 3)
print("a = ", a)
  
b = torch.full([2, 5], 3.5)
print("b = ", b)

Producción:

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

Publicación traducida automáticamente

Artículo escrito por sanskar27jain y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *