Fonction PyTorch torch.zeros_like

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.zeros_likeest une fonction de PyTorch utilisée pour créer un tenseur entièrement nul ayant la même forme que le tenseur d'entrée.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions