我有四个张量
> H(h,r)
> A(a,r)
> D(d,r)
> T(a,t,r)
对于a中的每个i,存在形状(t,r)的对应T [i].
pred = np.einsum('hr,ar,dr,tr ->hadt',H,A,D,T[0])
for i in range(a):
pred[:,i:i+1,:,:] = np.einsum('hr,tr ->HADT',A[i:i+1],T[i])
但是,我想在不使用for循环的情况下进行此计算.原因是我正在使用autograd,它目前不适用于项目分配!
最佳答案
一种方法是使用T的所有尺寸 –
np.einsum('Hr,Ar,Dr,ATr ->HADT',T)
因为,我们需要对所有输入中的轴-r进行求和,同时保留输出中的所有其他轴(轴),我没有看到任何中间方式这样做/在此基础上引入任何基于点的工具来利用BLAS .
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。