Fonction torch.ones_like de PyTorch

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

torch.ones_likeIl s'agit d'une fonction de PyTorch utilisée pour créer un tenseur de uns ayant la même forme que le tenseur d'entrée.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

x = torch.randn(2, 3)
y = torch.ones_like(x)

print(y)

Le résultat de sortie est:

tensor([[1., 1., 1.],
        [1., 1., 1.]])

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

Autres extensions