nvidia.dali.fn.reductions#

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

函数

设备支持

简短描述

reductions.max

CPU, GPU

获取沿指定轴的最大输入元素。

reductions.mean

CPU, GPU

获取沿指定轴的元素的平均值。

reductions.mean_square

CPU, GPU

获取沿指定轴的元素的均方值。

reductions.min

CPU, GPU

获取沿指定轴的最小输入元素。

reductions.rms

CPU, GPU

获取沿指定轴的元素的均方根值。

reductions.std_dev

CPU, GPU

获取沿指定轴的元素的标准差。

reductions.sum

CPU, GPU

获取沿指定轴的元素的总和。

reductions.variance

CPU, GPU

获取沿指定轴的元素的方差。