请问你代码中的gather_output_mlu = gather_mlu(data_mlu, shape_data, index_mlu, shape_index, CNML_DIM_C , cnrt_queue);函数定义哪儿可以找到?展开
Cambricon-CNML-Developer-Guide-EN-v7.10.7.7z还请参考cnml文档的·4.68节gather相关接口说明。
请问你代码中的gather_output_mlu = gather_mlu(data_mlu, shape_data, index_mlu, shape_index, CNML_DIM_C , cnrt_queue);函数定义哪儿可以找到?展开
老师我在pytorch中对W维进行gather,我尝试在cnml中按照NCHW的输入,设置CNML_DIM_C来欺骗芯片处理C维实际是W维。输入的index矩阵为NCHW的四维矩阵(但是我看cnml的官方手册中除了设定的dim外都要求是1维嘛?),最后的结果与pytorch的结果部分相同,我觉得还有我的理解问题。麻烦老师您帮我解答一下
请问你代码中的gather_output_mlu = gather_mlu(data_mlu, shape_data, index_mlu, shape_index, CNML_DIM_C , cnrt_queue);函数定义哪儿可以找到?展开
老师您好,函数定义在Inference_offline/Mygather.cpp at main · Sun2018421/Inference_offline · GitHub,非常感谢老师您的帮助
请登录后评论