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

Question

我有一个形状为[N，60,60,1]的4-D张量[batch_size，num_rows，num_cols，num_values]，我想用[60,60]矩阵加权WH维度。 我怎样才能做到这一点？

Answer 1

我有一个形状为[N，60,60,1]的4-D张量[batch_size，num_rows，num_cols，num_values]，我想用[60,60]矩阵加权WH维度。 我怎样才能做到这一点？

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

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