Función PyTorch torch.full_like


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

torch.full_likeEs una función en PyTorch que se utiliza para crear un tensor de relleno con la misma forma que el tensor de entrada.

Definición de la función

torch.full_like(input, fill_value, dtype=None, device=None, requires_grad=False)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor de entrada
input = torch.tensor([[1, 2], [3, 4]])

# Crear un tensor lleno de 7 con la misma forma que la entrada
x = torch.full_like(input, 7)

print(x)

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

Otras extensiones