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

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) )

这将broadcast沿着张量轴0的矩阵。