La Data Science

Valeur Aléatoire en PyTorch

Problème :

Comment générer un nombre flottant aléatoire avec PyTorch?

Solution :

Utiliser la fonction rand de torch.

torch.rand(size) renvoie un tenseur dont les dimensions sont spécifiées par size et dont les valeurs sont des flottants dans [0, 1[, choisis selon une loi uniforme.

Exemple :

import torch
x = torch.rand(1)
# génère un tenseur de dimension 1 et dont la valeur est choisie uniformément sur [0, 1[.
view raw random-float-torch.py hosted with ❤ by GitHub

Jean MP

,
,

Laisser un commentaire