Fonction torch.rand de PyTorch

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

torch.randC'est une fonction dans PyTorch utilisée pour créer des tenseurs aléatoires à partir d'une distribution uniforme [0, 1).

Ceci est couramment utilisé en apprentissage profond pour des scénarios tels que l'initialisation, la génération de masques aléatoires, etc.

Définition de fonction

torch.rand(*size, dtype=None, device=None, requires_grad=False)

Parameters:

  • *size(int) : Forme du tenseur.
  • dtype(torch.dtype, facultatif) : Type de données.
  • device(torch.device, facultatif) : Périphérique.
  • requires_grad(bool, facultatif) : Indique si le calcul du gradient est nécessaire.

Valeur de retour:

  • torch.Tensor: Renvoie un tenseur contenant des nombres aléatoires dans l'intervalle [0, 1).

Exemples d'utilisation

Exemple 1 : Créer un tenseur aléatoire

Exemple

import torch

# Créer un tenseur aléatoire 3x4
x = torch.rand(3, 4)

print(x)
print("Plage : [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))

Le résultat de sortie est:

tensor([[0.6272, 0.1802, 0.8336, 0.3900],
        [0.2188, 0.6593, 0.3598, 0.4021],
        [0.9247, 0.0144, 0.2587, 0.6634]])
范围: [0.0144, 0.9247)

Exemple 2 : Créer un masque aléatoire

Exemple

import torch

# Créer un masque aléatoire avec une probabilité de 50%
mask = torch.rand(10) > 0.5

print("Mask:", mask)
print("True count:", mask.sum().item())

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

Autres extensions