Функция torch.inner в PyTorch
Справочное руководство PyTorch torch
torch.innerЭто функция в PyTorch, используемая для вычисления внутреннего произведения двух тензоров. Для векторов она эквивалентна скалярному произведению; для многомерных тензоров она вычисляет внутреннее произведение по указанным измерениям.
Определение функции
torch.inner(input, other, out=None)
Parameters:
input(Tensor): первый входной тензор.other(Tensor): Второй входной тензор.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает результат внутреннего произведения.
Примеры использования
Пример - Внутреннее произведение векторов
import torch
# Создайте два вектора
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])
# Вычислить внутреннее произведение
result = torch.inner(a, b)
print("Вектор a:", a)
print("Вектор b:", b)
print("Результат внутреннего произведения:", result)
# Создайте два вектора
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])
# Вычислить внутреннее произведение
result = torch.inner(a, b)
print("Вектор a:", a)
print("Вектор b:", b)
print("Результат внутреннего произведения:", result)
Результат вывода:
向量 a: tensor([1., 2., 3.]) 向量 b: tensor([4., 5., 6.]) 内积结果: tensor(32.)
Пример - Внутреннее произведение матриц
import torch
# Создать две матрицы
A = torch.randn(2, 3)
B = torch.randn(2, 3)
# Вычислить скалярное произведение вдоль последнего измерения
result = torch.inner(A, B)
print("Форма A:", A.shape)
print("Форма B:", B.shape)
print("Форма результата:", result.shape)
# Создать две матрицы
A = torch.randn(2, 3)
B = torch.randn(2, 3)
# Вычислить скалярное произведение вдоль последнего измерения
result = torch.inner(A, B)
print("Форма A:", A.shape)
print("Форма B:", B.shape)
print("Форма результата:", result.shape)
Другие расширения