Función PyTorch torch.zeros_like

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

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

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

x = torch.randn(3, 4)
y = torch.zeros_like(x)

print(Forma de x:, x.shape)
print(Forma de y:, y.shape)
print(y)

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

Otras extensiones