自定义操作# 在 C++ 中创建自定义操作符 先决条件 操作符定义 CPU 操作符实现 GPU 操作符实现 构建插件 导入插件 预加载插件 Python 操作符 定义操作 定义 Pipeline 运行 Pipeline 并可视化结果 各种 Python 操作符 使用 Python 操作符处理 GPU 数据 CuPy 操作 定义 Pipeline 运行 Pipeline 并可视化结果 高级：DLTensorPythonFunction 中的设备同步 Numba 函数 - 运行编译后的 C 回调函数 定义交换宽度和高度的形状函数 定义基于输入样本填充输出样本的处理函数 定义 Pipeline 在 DALI 中运行自定义 JAX 增强 设置示例 添加使用 JAX 定义的增强 JAX 函数转换 多个输入和输出 常规 pipeline 中的 JAX 增强 在多个 GPU 上运行 JAX 增强 设置分布式运行 带有分片的迭代器 带有分片的 JAX 增强
