PyTorch torch.ones_like Funktion
PyTorch torch Referenzhandbuch
torch.ones_likeEs ist eine Funktion in PyTorch, mit der ein Tensor aus lauter Einsen mit derselben Form wie der Eingabetensor erstellt wird.
Funktionsdefinition
torch.ones_like(input, dtype=None, device=None, requires_grad=False)
Anwendungsbeispiel
Beispiel
import torch
x = torch.randn(2, 3)
y = torch.ones_like(x)
print(y)
x = torch.randn(2, 3)
y = torch.ones_like(x)
print(y)
Das Ausgabeergebnis ist:
tensor([[1., 1., 1.],
[1., 1., 1.]])
Andere Erweiterungen