我是新手Einsum
并且想要一个特殊情况 -einsum
用于将矩阵的所有元素相互相乘;说给定一个2D
矩阵:-
np.random.rand((16,2))
跨轴乘以元素获得(16,)
然后再次将它们相乘以获得(1,)
标量。这类似于:-
[[1, 2], ==> [2, 12] ==> 24
[3, 4]]
我尝试使用类似的东西: -
...("ij->")
但这绝对不是我想要的,因为它不是乘法
我们怎么能用 编写这么简单的操作einsum
呢?einsum 不能处理每一个操作,而是针对一些困难情况进行优化吗?