Función PyTorch torch.inner

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.innerEs una función en PyTorch para calcular el producto interno de dos tensores. Para vectores, equivale al producto escalar; para tensores de alta dimensión, calcula el producto interno en la dimensión especificada.

Definición de la función

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

Parámetros:

  • input(Tensor): El primer tensor de entrada.
  • other(Tensor): El segundo tensor de entrada.
  • out(Tensor, opcional): El tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el resultado del producto interno.

Ejemplo de uso

Ejemplo - Producto interno de vectores

import torch

# Crear dos vectores
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])

# Calcular el producto interno
result = torch.inner(a, b)

print("Vector a:", a)
print("Vector b:", b)
print("Resultado del producto interno:", result)

El resultado de salida es:

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

Ejemplo - Producto interno de matrices

import torch

# Crear dos matrices
A = torch.randn(2, 3)
B = torch.randn(2, 3)

# Calcular el producto interno en la última dimensión
result = torch.inner(A, B)
print("Forma de A:", A.shape)
print("Forma de B:", B.shape)
print("Forma del resultado:", result.shape)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones