样例总结
In-place算子在入图过程中存在一些注意事项,在实际操作时请注意:
- Tensor类型输入在前,基本类型输入在后,不要返回任何被修改的输入。
- 以算子工程方式开发PyTorch算子的Ascend C实现,Ascend C算子名保证为PyTorch算子的大驼峰格式,同时入参顺序、类型与PyTorch算子一致。Ascend C算子原型定义时,所有被修改的输入需要添加与其同名的输出。
- 注册对应的非In-place算子时,算子名为“In-place算子名”+"_functional"后缀。
父主题: In-place算子开发和入图样例