Pytorch – 使用torch.matmul()替换torch.einsum(‘nctw,cd->ndtw’,(a,b))算子模式
1 pytorch的torch.matmul()函数 函数形式 torch.matmul(input, other, *, out=None) → Tensor 该函数主要是用于求解两个tensor的矩阵乘积。 该函数根据输入的两个tensor的维度的不同进行不一样的张量运算，如下所示 如果两个参数…
- Pytorch
- 2022-03-29
cd
1篇文章
资助我们
帮助StubbornHuang Blog继续发展下去
用支付宝或者微信扫描二维码
对站长进行赞助
金额随意，礼轻义重