nvidia.dali.fn.reductions#
下表列出了 nvidia.dali.fn.reductions
模块中所有可用的操作
函数 |
设备支持 |
简短描述 |
---|---|---|
CPU, GPU |
获取沿指定轴的最大输入元素。 |
|
CPU, GPU |
获取沿指定轴的元素的平均值。 |
|
CPU, GPU |
获取沿指定轴的元素的均方值。 |
|
CPU, GPU |
获取沿指定轴的最小输入元素。 |
|
CPU, GPU |
获取沿指定轴的元素的均方根值。 |
|
CPU, GPU |
获取沿指定轴的元素的标准差。 |
|
CPU, GPU |
获取沿指定轴的元素的总和。 |
|
CPU, GPU |
获取沿指定轴的元素的方差。 |