Fonction torch.normal de PyTorch

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

torch.normalest une fonction dans PyTorch utilisée pour générer des nombres aléatoires à partir d'une distribution normale.

Définition de fonction

torch.normal(mean, std, *, generator=None, out=None)
torch.normal(mean, std, size, *, generator=None, out=None)

Description du paramètre

  • mean- Moyenne de la distribution normale
  • std- Écart-type de la distribution normale
  • size- Forme du tenseur de sortie (facultatif)
  • generator- Générateur de nombres aléatoires (facultatif)
  • out- Tenseur de sortie (facultatif)

Exemple d'utilisation

Exemple

import torch

# Générer un tenseur aléatoire en utilisant une moyenne et un écart-type fixes
result1 = torch.normal(mean=0.0, std=1.0, size=(3, 3))
print("Tenseur aléatoire de loi normale 3x3 de moyenne 0 et d'écart-type 1 :")
print(result1)

# Utiliser des tenseurs comme moyenne et écart-type
mean = torch.tensor([0.0, 1.0, 2.0])
std = torch.tensor([1.0, 2.0, 3.0])
result2 = torch.normal(mean, std)
print("nRésultat d'échantillonnage en utilisant des paramètres tenseur :")
print(result2)

# Utiliser une moyenne scalaire et un écart-type sous forme de tenseur
result3 = torch.normal(mean=0.0, std=torch.tensor([1.0, 2.0, 3.0]))
print("nRésultat d'échantillonnage avec une moyenne scalaire et un écart-type sous forme de tenseur :")
print(result3)

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

Autres extensions