torch.chain_matmul

torch.chain_matmul(*matrices, out=None) [исходный код]

Возвращает произведение матриц $N$ двумерных тензоров. Это произведение эффективно вычисляется с использованием алгоритма порядка цепочки матриц, который выбирает порядок с наименьшей вычислительной стоимостью ([CLRS]). Обратите внимание: поскольку эта функция вычисляет произведение, значение $N$ должно быть больше или равно 2; если оно равно 2, возвращается тривиальное произведение двух матриц. Если $N$ равно 1, функция ничего не делает — исходная матрица возвращается без изменений.

Предупреждение

torch.chain_matmul() устарела и будет удалена в одном из будущих выпусков PyTorch. Вместо неё используйте torch.linalg.multi_dot(), которая принимает список из двух или более тензоров вместо нескольких аргументов.

Параметры:

• matrices (Тензоры...) – последовательность из двух или более двумерных тензоров, произведение которых нужно вычислить.
• out (Тензор, необязательно) – выходной тензор. Игнорируется, если out = None.

Возвращает:

если размерности $i^{th}$ тензора равны $p_{i} \times p_{i + 1}$, то размерности произведения будут равны $p_{1} \times p_{N + 1}$.

Тип возвращаемого значения:

Тензор

Пример:

>>> a = torch.randn(3, 4)
>>> b = torch.randn(4, 5)
>>> c = torch.randn(5, 6)
>>> d = torch.randn(6, 7)
>>> # will raise a deprecation warning
>>> torch.chain_matmul(a, b, c, d)
tensor([[ -2.3375,  -3.9790,  -4.1119,  -6.6577,   9.5609, -11.5095,  -3.2614],
        [ 21.4038,   3.3378,  -8.4982,  -5.2457, -10.2561,  -2.4684,   2.7163],
        [ -0.9647,  -5.8917,  -2.3213,  -5.2284,  12.8615, -12.2816,  -2.5095]])