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()), в результате чего тензор вывода имеет на 1 меньше измерение, чем 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])