Функция torch.ger в PyTorch

Pytorch torch 参考手册 Справочное руководство Pytorch torch

torch.gerЭто функция в PyTorch, используемая для вычисления внешнего произведения (матрицы Грама) двух векторов.

Определение функции

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

Пример использования

Пример

import torch

# Создайте два вектора
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])

# Вычислить внешнее произведение
result = torch.ger(vec1, vec2)
print("Outer product result:")
print(result)

Результат вывода:

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

Pytorch torch 参考手册 Справочное руководство Pytorch torch

Другие расширения