Каков эффективный способ умножения цепочки тензоров в тензорном потоке

Question

У меня есть 3 разреженных тензора размеров A = P * N, B = Q * N и C = R * N. Каков эффективный способ вычисления матрицы произведения A * B * C такой, что размерность матрицы произведения P * Q * R в тензорном потоке.

Я попытался с tf.matmul и за ним следует tf.reshape, но он не даст матрицу продукта с размером, указанным выше.

Спасибо.

Answer 1

tf.einsum Должно вам помочь. Хотя я не совсем уверен, что вы подразумеваете под A*B*C, поскольку размеры несовместимы для матричного умножения. Возможно, что-то вроде этого:

R = tf.einsum('il,jl,kl->ijk', A, B, C)