nvidia.dali.plugin.jax.fn

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

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