Функция torch.mul в PyTorch
Справочное руководство PyTorch torch
torch.mulЭто функция в PyTorch, используемая для поэлементного умножения. Она умножает два тензора или тензор и скаляр.
Это одна из самых фундаментальных математических операций в глубоком обучении.
Определение функции
torch.mul(input, other, out=None)
Parameters:
input(Tensor): первый входной тензор.other(Tensor или float): второй входной тензор или скаляр.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает умноженный тензор.
Примеры использования
Пример 1: Поэлементное умножение тензоров
Пример
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
c = torch.mul(a, b)
print(c)
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
c = torch.mul(a, b)
print(c)
Результат вывода:
tensor([ 4, 10, 18])
Пример 2: Умножение на скаляр
Пример
import torch
a = torch.tensor([1, 2, 3])
# Умножить на 2
b = torch.mul(a, 2)
print(b)
# Также можно использовать оператор *
c = a * 2
print(c)
a = torch.tensor([1, 2, 3])
# Умножить на 2
b = torch.mul(a, 2)
print(b)
# Также можно использовать оператор *
c = a * 2
print(c)
Результат вывода:
tensor([2, 4, 6]) tensor([2, 4, 6])
Другие расширения