tf.tensordot

Свёртка тензоров a и b по заданным осям и внешнее произведение.

tf.tensordot(
    a, b, axes, name=None
)

Tensordot (также известная как свёртка тензоров) суммирует произведения элементов a и b по индексам, заданным в axes.

Эта операция соответствует numpy.tensordot(a, b, axes).

Пример 1. Если a и b — матрицы (порядка 2), случай axes=1 эквивалентен умножению матриц.

Пример 2. Если a и b — матрицы (порядка 2), случай axes = [[1], [0]] эквивалентен умножению матриц.

Пример 3. Если a и b — матрицы (порядка 2), случай axes=0 даёт внешнее произведение — тензор порядка 4.

Пример 4. Пусть \(a_{ijk}\) и \(b_{lmn}\) обозначают два тензора порядка 3. Тогда contract(a, b, [[0], [2]]) — это тензор порядка 4 \(c_{jklm}\), элемент которого с индексами \((j,k,l,m)\) задаётся формулой:

\( c_{jklm} = \sum_i a_{ijk} b_{lmi} \).

В общем случае order(c) = order(a) + order(b) - 2*len(axes[0]).