x2paddle算子不支持怎么办？

Open AIHHU opened this issue 4 years ago • 1 comments

感谢您参与 X2Paddle 社区！问题模版为了 X2Paddle 能更好的迭代，例如新功能发布、 RoadMaps 和错误跟踪. :smile_cat:

问题描述

ZVW4Y3NL1SO5~HLL2`$_60C

请问有办法自定义算子相关的教程吗？

Apr 13 '22 05:04 AIHHU

@AIHHU 目前aten::linear已在这个PR实现：https://github.com/PaddlePaddle/X2Paddle/pull/769

关于aten::atan2以及aten::group_norm我看paddle和torch都是相同功能一一对应的关系，对应paddle.atan2以及paddle.nn.GroupNorm

可以在x2paddle/op_mapper/pytorch2paddle/aten.py中

1、参照aten_tanh这个函数，补充aten_atan2函数

2、参照aten_batch_norm补充aten_group_norm函数

期待您的PR~

Apr 13 '22 07:04 wjj19950828