Función torch.outer de PyTorch

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

torch.outerEs una función en PyTorch que se utiliza para calcular el producto exterior de dos vectores (también llamado producto cruz o producto tensorial). El producto exterior multiplica cada elemento de un vector por todos los elementos del otro vector, generando una matriz.

Definición de la función

torch.outer(vec1, vec2, out)

Ejemplo de uso

Ejemplo

import torch

# Definir dos vectores
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])

# Calcular el producto exterior
result = torch.outer(vec1, vec2)

print(result)

El resultado de salida es:

tensor([[ 4,  5,  6],
        [ 8, 10, 12],
        [12, 15, 18]])

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

Otras extensiones