Fonction torch.ones de PyTorch

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

torch.onesest une fonction de PyTorch utilisée pour créer un tenseur rempli de 1. Elle crée un tenseur de la forme spécifiée avec tous les éléments initialisés à 1.

Ceci est souvent utilisé en apprentissage profond pour initialiser certains paramètres, créer des masques ou servir de valeurs de référence pour des opérations mathématiques.

Définition de fonction

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

Parameters:

  • *size(int) : La forme du tenseur, par exemple,3、(3, 4)、(2, 3, 4)etc.
  • dtype(torch.dtype, facultatif) : Spécifie le type de données du tenseur, par défauttorch.float32。
  • device(torch.device, facultatif) : Spécifie le périphérique où le tenseur est stocké.
  • requires_grad(bool, optionnel) : Indique s'il faut calculer les gradients.
  • pin_memory(bool, facultatif) : Indique s'il faut utiliser la mémoire épinglée.

Valeur de retour:

  • torch.Tensor: Renvoie un tenseur rempli de 1.

Exemples d'utilisation

Exemple 1 : Créer un tenseur 1D rempli de 1

Exemple

import torch

# Créer un tenseur rempli de 1 contenant 5 éléments
x = torch.ones(5)

print(x)

Le résultat de sortie est:

tensor([1., 1., 1., 1., 1.])

Exemple 2 : Créer un tenseur 2D rempli de 1

Exemple

import torch

# Créer un tenseur 3x4 rempli de 1 (matrice)
x = torch.ones(3, 4)

print(x)
print(x.shape)

Le résultat de sortie est:

tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
torch.Size([3, 4])

Exemple 3 : Créer un tenseur de uns 3D

Exemple

import torch

# Créer un tenseur de uns 2x3x4
x = torch.ones(2, 3, 4)

print(x.shape)

Le résultat de sortie est:

torch.Size([2, 3, 4])

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

Autres extensions