Einsum 问答列表

如何获得 (n,1,3) numpy 数组中每个数组的点积,以及 (n,3) 数组的所有数组,并得到形状为 (n,n) 的结果数组?

作者:Devarapalli Vamsi 提问时间:9/5/2023

我尝试了 einsum np.einsum,但这给了我错误,输出不能重复相同的字母。 np.einsum('aij,aj->aa', vector1, vector2) 还尝试了 np.dot ...

此代码将循环矩阵乘法转换为 einsum 是否正确?

作者:Sai Kishore 提问时间:11/17/2023

我已经为我想执行的矩阵乘法编写了一个基于 for 循环和基于 einsum 的代码。你能帮我检查它的正确性吗? ` w = torch.randn((10,32,32)) x = torch.ra...


共2条 当前第1页