nvidia.dali.fn.reductions.rms¶
-
nvidia.dali.fn.reductions.
rms
(*inputs, **kwargs)¶ Gets root mean square of elements along provided axes.
- Supported backends
‘cpu’
‘gpu’
- Parameters
input (TensorList) – Input to the operator.
- Keyword Arguments
axes (int or list of int, optional) –
Axis or axes along which reduction is performed.
Not providing any axis results in reduction of all elements.
axis_names (layout str, optional) –
Name(s) of the axis or axes along which the reduction is performed.
The input layout is used to translate the axis names to axis indices, for example
axis_names="HW"
with input layout “FHWC” is equivalent to specifyingaxes=[1,2]
. This argument cannot be used together withaxes
.bytes_per_sample_hint (int or list of int, optional, default = [0]) –
Output size hint, in bytes per sample.
If specified, the operator’s outputs residing in GPU or page-locked host memory will be preallocated to accommodate a batch of samples of this size.
dtype (nvidia.dali.types.DALIDataType, optional, default = DALIDataType.NO_TYPE) – Output data type. This type is used to accumulate the result.
keep_dims (bool, optional, default = False) – If True, maintains original input dimensions.
preserve (bool, optional, default = False) – Prevents the operator from being removed from the graph even if its outputs are not used.
seed (int, optional, default = -1) –
Random seed.
If not provided, it will be populated based on the global seed of the pipeline.