如何将 4-D 张量乘以 2-D 矩阵?

Question

我有一个形状为 [N，60,60,1] 的 4-D 张量 [batch_size、num_rows、num_cols、num_values]，我想用 [60,60] 矩阵对 WH 维度进行加权。 我该怎么做？

Answer 1

我有一个 4-D 张量 [batch_size、num_rows、num_cols、num_values] 形状 [N，60,60,1]，我想用 [60,60] 矩阵。我该怎么做？

tensor * matrix.reshape( (1, 60, 60, 1) )

这将沿张量的轴 0 广播矩阵。