问如何编写张量积形式的einsum运算：'i，jk->ijk‘？
EN

Stack Overflow用户

提问于 2020-11-10 18:50:48

回答 1查看 53关注 0票数 0

我正在使用pytorch，并希望将一维(特征)向量广播到三维张量(特征×高度×宽度)，以使用卷积神经网络来调节图像的编码。如此Deepmind paper的图S1所示。现在我使用的是：

# features: 1D vector
f = torch.einsum('i,jk->ijk',features,torch.ones([5,5]))

回答已采纳

发布于 2020-11-11 02:01:25

使用广播语义，您也可以使用以下命令进行计算

f = features.reshape(-1, 1, 1) * torch.ones(1, 5, 5)

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/64767467

复制

相似问题

问如何编写张量积形式的einsum运算：'i，jk->ijk‘？EN