Fonction torch.rand_like de PyTorch
Manuel de référence PyTorch torch
torch.rand_likeest une fonction dans PyTorch utilisée pour créer un tenseur aléatoire uniformément distribué ayant la même forme que le tenseur d'entrée.
Définition de fonction
torch.rand_like(input, dtype=None, device=None, requires_grad=False)
Exemples d'utilisation
Exemple
import torch
# Créer un tenseur d'entrée
input = torch.tensor([[1, 2], [3, 4]])
# Créer un tenseur aléatoire uniformément distribué de même forme que l'entrée [0, 1)
x = torch.rand_like(input)
print(x)
# Créer un tenseur d'entrée
input = torch.tensor([[1, 2], [3, 4]])
# Créer un tenseur aléatoire uniformément distribué de même forme que l'entrée [0, 1)
x = torch.rand_like(input)
print(x)
Autres extensions