python – 矢量化numpy.einsum
发布时间:2020-12-16 22:34:01 所属栏目:Python 来源:网络整理
导读:我有四个张量 H(h,r) A(a,r) D(d,r) T(a,t,r) 对于a中的每个i,存在形状(t,r)的对应T [i]. 我需要做一个np.einsum来产生以下结果(pred): 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',
我有四个张量 > H(h,r) 对于a中的每个i,存在形状(t,r)的对应T [i]. 我需要做一个np.einsum来产生以下结果(pred):
但是,我想在不使用for循环的情况下进行此计算.原因是我正在使用autograd,它目前不适用于项目分配! 最佳答案
一种方法是使用T的所有尺寸 –
因为,我们需要对所有输入中的轴-r进行求和,同时保留输出中的所有其他轴(轴),我没有看到任何中间方式这样做/在此基础上引入任何基于点的工具来利用BLAS . (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |