在训练一个知识图谱模型时，需要用到系数矩阵的乘法，目前在CPU平台上代码已通。但该API目前仅支持CPU和GPU，当把运行环境设为Ascend时就报数据同步错误。目前的任务需求是要在Ascend910上运行，所以请问是否可以在Ascend环境下使用该API进行稀疏矩阵乘法，有没有什么解决办法？