替换

1篇文章
1 问题:TensorRT暂时未实现einsum算子 在ST-GCN中使用了爱因斯坦求和算子torch.einsum, def forward(self, x, A): assert A.size(0…

关注我们的公众号

微信公众号
 
站点已勉强运行: