【功能模块】
ubuntu 20.04 python3.9 cuda11.1环境下mindspore 1.7版本,Tensor索引功能速度较慢
【操作步骤&问题现象】
1、在mindspore 1.7的GPU工作模式下,对Tensor的索引速度较慢,和python的np.array格式的索引速度比较,mindspore的Tensor索引速度慢了好几个数量级
【截图信息】
您好 经过这边复现 目前确实存在Tensor索引速度过慢的问题 因为目前mindspore的Tensor索引过程中需要调用maketuple算子来将Gather的输出变成一个tuple 相对会比较耗时 我们会在Q4阶段对算子性能进行集中整改 感谢您的使用。