Функция PyTorch torch.prod

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

torch.prod— это функция в PyTorch, используемая для возврата произведения всех элементов тензора.

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

torch.prod(input, dim, keepdim=False)

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

Пример

import torch

x = torch.tensor([1, 2, 3, 4])

# Возвращает произведение всех элементов
print("Произведение всех элементов:", torch.prod(x))

# Произведение по dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("произведение по dim=0:", torch.prod(y, dim=0))
print("произведение по dim=1:", torch.prod(y, dim=1))

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

所有元素乘积: tensor(24)
dim=0 乘积: tensor([4, 10, 18])
dim=1 乘积: tensor([  6, 120])

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

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