有关einsum,两指定的维度进行相乘，没指定就求和。
python X = np.array([[[1, 5], [2, 6]], [[3, 7], [4, 8]]]) A = np.array([[1, 2], [3, 4], [5, 6]]) >>> X array([[[1, 5], [2, 6]], [[3, 7], [4, 8]]]) >>> A array([[1, 2], [3, 4], [5, 6]]) >>> X.shape (2, 2, 2) >>> A.shape (3, 2) >>> Y = np.einsum('kij, lk -> lij', X, A) >>> Y.shape (3, 2, 2) >>> Y array([[[ 7, 19], [10, 22]], [[15, 43], [22, 50]], [[23, 67], [34, 78]]]) 计算过程如下
图片转不过来凑活看吧
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/220996.html原文链接：https://javaforall.net
