当前位置：网站首页>Pytorch中的torch.index_select对应MindSpore哪个方法

Pytorch中的torch.index_select对应MindSpore哪个方法

2022-08-10 03:28:00 【小乐快乐】

mindspore中的哪个API函数可以替换Pytorch中的torch.index_select函数呢？现在正在做minds pore的模型迁移

pytorch被替换的函数如下截图所示：

【截图信息】

mindspore.ops.Gather — MindSpore master documentation

原网站

版权声明
本文为[小乐快乐]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_45666880/article/details/126249552

边栏推荐

猜你喜欢

随机推荐