Fonction torch.inner de PyTorch

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

torch.innerIl s'agit d'une fonction de PyTorch utilisée pour calculer le produit intérieur de deux tenseurs. Pour les vecteurs, elle est équivalente au produit scalaire ; pour les tenseurs de dimension supérieure, elle calcule le produit intérieur selon les dimensions spécifiées.

Définition de fonction

torch.inner(input, other, out=None)

Parameters:

  • input(Tensor) : Le premier tenseur d'entrée.
  • other(Tensor) : Le deuxième tenseur d'entrée.
  • out(Tensor, optionnel) : Le tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le résultat du produit intérieur.

Exemples d'utilisation

Exemple - Produit intérieur de vecteurs

import torch

# Créer deux vecteurs
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])

# Calculer le produit intérieur
result = torch.inner(a, b)

print("Vecteur a :", a)
print("Vecteur b :", b)
print("Résultat du produit intérieur :", result)

Le résultat de sortie est:

向量 a: tensor([1., 2., 3.])
向量 b: tensor([4., 5., 6.])
内积结果: tensor(32.)

Exemple - Produit intérieur de matrices

import torch

# Créer deux matrices
A = torch.randn(2, 3)
B = torch.randn(2, 3)

# Calculer le produit scalaire le long de la dernière dimension
result = torch.inner(A, B)
print("Forme de A:", A.shape)
print("Forme de B:", B.shape)
print("Forme du résultat :", result.shape)

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

Autres extensions