numpy, Tensorflow一个矩阵与多维矩阵相乘(不一样维度的Tensor相乘)

张量相乘在深度学习中常常遇到,这里咱们简单介绍一下numpy和tensorflow中的张量相乘。python 一、对于维数同样的张量相乘。A.shape =(b,m,n), B.shape = (b,n,k).dom numpy.matmul(A,B) 结果shape为(b,m,k)学习 这里要求第一维度相同。后两个维度能知足矩阵相乘条件。code 二、可是对于维数不同的张量如何处理呢?好比 A.
相关文章
相关标签/搜索