Función torch.full de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.fullEs una función de PyTorch que se utiliza para crear un tensor relleno con el valor especificado.

Definición de la función

torch.full(size, fill_value, dtype=None, device=None, requires_grad=False)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor de 3x4 con valor de relleno 7
x = torch.full((3, 4), 7)

print(x)

El resultado de salida es:

tensor([[7., 7., 7., 7.],
        [7., 7., 7., 7.],
        [7., 7., 7., 7.]])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones