Функция PyTorch torch.outer

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

torch.outer— это функция в PyTorch, используемая для вычисления внешнего произведения (также называемого векторным произведением или тензорным произведением) двух векторов. Внешнее произведение умножает каждый элемент одного вектора на все элементы другого вектора, создавая матрицу.

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

torch.outer(vec1, vec2, out)

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

Пример

import torch

# Определить два вектора
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])

# Вычислить внешнее произведение
result = torch.outer(vec1, vec2)

print(result)

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

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

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

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