Матрица Tensorflow SxN умножается на матрицу SxD для вывода массива SxNxD

ElleryL спросил: 07 октября 2018 в 12:15 в: python

У меня есть матрица SxN, где каждый K_ij является скаляром

[[K_11,K_12,K_13,..,K_1N],
 [K_21,K_22,K_23,..,K_2N],
 ..,
 [K_S1,K_S2,K_S3,..,K_SN]]

Другая матрица SxD такая, что каждый X_i является вектором Dx1

[X_1,
 X_2,
 ...,
 X_S]

Мне нужны такие результаты, которые дают мне

[[K_11 * X_1],[K_12 * X_1] ..., [K_1N * X_1],
 [K_21 * X_2],[K_22 * X_2] ..., [K_2N * X_2],
 [K_31 * X_3],[K_32 * X_2] ..., [K_3N * X_3],
  ......,
  [K_S1 * X_S],[K_S2 * X_S] ..., [K_SN * X_S]]

где означает умножение

Есть ли эффективный способ сделать это?

0 ответов