整理了一下我自己使用TensorFlow中常用的一些运算操作:
1. 多维张量计算
两个三维矩阵的乘法怎样计算呢?tensorflow把前面的维度当成是batch,对最后两维进行普通的矩阵乘法。也就是说,最后两维之前的维度,都需要相同
比如说:a = [2, 3, 2] , b = [2, 2, 3] 其实可以理解a矩阵的两个3*2的矩阵对应乘以b矩阵的两个2*3的矩阵
参考:https://blog.youkuaiyun.com/weixin_42445581/article/details/82791811