Función PyTorch torch.ger

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

torch.gerEs una función en PyTorch para calcular el producto exterior (matriz de Gram) de dos vectores.

Definición de la función

torch.ger(input, vec2, out=None)

Ejemplo de uso

Ejemplo

import torch

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

# Calcular el producto exterior
result = torch.ger(vec1, vec2)
print("Resultado del producto exterior:")
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