tensorflow2再做矩阵相乘的时候如果维度有None,则矩阵相乘结果不对。
也不知道为什么,只能用矩阵对应元素相乘在相加的形式来实现矩阵相乘,这个困扰了我好几天,所以记下来。

inputs_tile@self.kernel

换成

tf.reduce_sum(inputs_expand*self.kernel,axis=-2)

相关文章:

  • 2022-01-22
  • 2021-12-15
  • 2021-09-07
  • 2021-11-01
  • 2022-02-07
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2021-04-27
  • 2021-06-07
  • 2021-12-11
相关资源
相似解决方案