Fonction torch.arange de PyTorch

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

torch.arangeest une fonction de PyTorch utilisée pour créer des tenseurs de progression arithmétique. Elle crée un tenseur unidimensionnel contenant une séquence arithmétique de la valeur de début à la valeur de fin.

Ceci est souvent utilisé en apprentissage profond pour créer des indices, des tableaux de plages, ou générer des séquences dans des boucles.

Définition de fonction

torch.arange(start=0, end, step=1, dtype=None, device=None, requires_grad=False)

Parameters:

  • start(float, optionnel) : La valeur de départ de la séquence, par défaut 0.
  • end(float) : La valeur de fin de la séquence (exclue).
  • step(float, optionnel) : La taille du pas, par défaut 1.
  • dtype(torch.dtype, optionnel) : Spécifie le type de données du tenseur.
  • device(torch.device, optionnel) : Spécifie le périphérique sur lequel le tenseur est stocké.
  • requires_grad(bool, optionnel) : Indique s'il faut calculer les gradients.

Valeur de retour:

  • torch.Tensor: Renvoie un tenseur unidimensionnel.

Exemples d'utilisation

Exemple 1 : De 0 à 5

Exemple

import torch

# Créer une séquence arithmétique de 0 à 4
x = torch.arange(5)

print(x)

Le résultat de sortie est:

tensor([0, 1, 2, 3, 4])

Exemple 2 : Spécifier les valeurs de début et de fin

Exemple

import torch

# Créer une séquence arithmétique de 2 à 8
x = torch.arange(2, 9)

print(x)

Le résultat de sortie est:

tensor([2, 3, 4, 5, 6, 7, 8])

Exemple 3 : Spécifier la taille du pas

Exemple

import torch

# Créer une séquence arithmétique de 0 à 10 avec un pas de 2
x = torch.arange(0, 11, 2)

print(x)

Le résultat de sortie est:

tensor([0, 2, 4, 6, 8, 10])

Exemple 4 : Taille du pas négative

Exemple

import torch

# Créer une séquence arithmétique de 10 à 0 avec un pas de -2
x = torch.arange(10, 0, -2)

print(x)

Le résultat de sortie est:

tensor([10,  8,  6,  4,  2])

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

Autres extensions