Función PyTorch torch.normal

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.normalEs una función en PyTorch utilizada para generar números aleatorios a partir de una distribución normal.

Definición de la función

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

Descripción de parámetros

  • mean- Media de la distribución normal
  • std- Desviación estándar de la distribución normal
  • size- Forma del tensor de salida (opcional)
  • generator- Generador de números aleatorios (opcional)
  • out- Tensor de salida (opcional)

Ejemplo de uso

Ejemplo

import torch

# Generar un tensor aleatorio usando una media y desviación estándar fijas
result1 = torch.normal(mean=0.0, std=1.0, size=(3, 3))
print("Tensor aleatorio 3x3 de distribución normal con media 0, desviación estándar 1:")
print(result1)

# Usar tensores como media y desviación estándar
mean = torch.tensor([0.0, 1.0, 2.0])
std = torch.tensor([1.0, 2.0, 3.0])
result2 = torch.normal(mean, std)
print("nResultado de muestreo usando parámetros de tensor:")
print(result2)

# Usar media escalar y desviación estándar de tensor
result3 = torch.normal(mean=0.0, std=torch.tensor([1.0, 2.0, 3.0]))
print("nResultado de muestreo con media escalar y desviación estándar de tensor:")
print(result3)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones