[ torch 参数更多 ]torch.utils.cpp_extension.CppExtension

torch.utils.cpp_extension.CppExtension

torch.utils.cpp_extension.CppExtension(name,
                                    sources,
                                    *args,
                                    **kwargs)

paddle.utils.cpp_extension.CppExtension

paddle.utils.cpp_extension.CppExtension(sources,
                                    *args,
                                    **kwargs)

Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | name | - | 参数 name,Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 | | sources | sources | 用于指定自定义 OP 对应的源码文件。 | |*args | *args | 用于指定 Extension 的其他参数,支持的参数与 setuptools.Extension 一致。 | | **kwargs | **kwargs | 用于指定 Extension 的其他参数,支持的参数与 setuptools.Extension 一致。 |