Tenseur

Un tenseur est un tableau multidimensionnel de nombres, généralisation des vecteurs (1 dimension) et matrices (2 dimensions). C'est la structure de données de base de PyTorch, avec un type (dtype), une forme (shape) et un périphérique (CPU ou GPU). Il peut enregistrer les opérations pour calculer des gradients.

Exemple

x = torch.tensor([[1., 2.], [3., 4.]])  # shape (2, 2)

Notions voisines

Tout le glossaire