nvidia.dali.plugin.jax.fn

下表列出了 nvidia.dali.plugin.jax.fn 模块中所有可用的操作

函数	设备支持	简短描述
jax_function	CPU, GPU	将处理 jax.Array 的函数转换为 DALI 操作符