torch.prod

torch.prod(input, *, dtype=None) → Tensor

Возвращает произведение всех элементов в тензоре input.

Параметры

input (Тензор) – входной тензор.

Ключевые аргументы

dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. Если указано, входной тензор приводится к типу dtype перед выполнением операции. Это полезно для предотвращения переполнения типов данных. По умолчанию: None.

Пример:

>>> a = torch.randn(1, 3)
>>> a
tensor([[-0.8020,  0.5428, -1.5854]])
>>> torch.prod(a)
tensor(0.6902)

torch.prod(input, dim, keepdim=False, *, dtype=None) → Тензор

Возвращает произведение каждой строки тензора input в заданном измерении dim.

Если keepdim равно True, тензор вывода имеет тот же размер, что и input, за исключением измерения dim, где он имеет размер 1. В противном случае dim сжимается (см. torch.squeeze()), в результате чего тензор вывода имеет на одну размерность меньше, чем input.

Параметры

• input (Тензор) – входной тензор.
• dim (int) – измерение для сокращения.
• keepdim (bool) – сохранять ли размерность вывода.

Ключевые аргументы

dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. Если указано, входной тензор приводится к типу dtype перед выполнением операции. Это полезно для предотвращения переполнения типов данных. По умолчанию: None.

Пример:

>>> a = torch.randn(4, 2)
>>> a
tensor([[ 0.5261, -0.3837],
        [ 1.1857, -0.2498],
        [-1.1646,  0.0705],
        [ 1.1131, -1.0629]])
>>> torch.prod(a, 1)
tensor([-0.2018, -0.2962, -0.0821, -1.1831])