使用torch普通算子组合替代torch.einsum爱因斯坦求和
具体来说,这个操作的输入是两个形状为 [b, n, d] 和 [b, m, d] 的张量 x 和 y,输出是一个形状为 [b, n, m] 的张量 z。其计算过程可以理解为:对于每个 b,z[b, n, m] 等于 x[b, n, :] 和 y[b, m, :] 之间的点积。为了用普通的 torch 操作符来替代 einsum,我们可以通过 torch.matmul 函数实现。torch.einsum(‘bnd, bmd->bnm’, x, y) 表示的是对张量 x 和 y 进行特定的求和和维度变换。
原创
2024-08-06 16:40:27 ·
148 阅读 ·
0 评论