我在使用mindspore复现网络的时候,使用了sklearn.neighbors.NearestNeighbors.kneighbors()的算子来计算tensor,数据的输入类型如图所示
计算速度非常慢,大大影响了网络的训练速度,请问该如何解决
注意到你图片中数据类型是fp32请问你的执行是在ascend上执行如果是的话ascend fp32支持度不高会导致执行速度比较慢;https://www.mindspore.cn/tutorials/experts/zh-CN/r1.8/others/mixed_precision.html?highlight=æ··å精度 可以参照上述链接转用混精进行运行