cr.nimble.dsp.scale_0_mean_1_var

cr.nimble.dsp.scale_0_mean_1_var(data, axis=- 1)[source]

Normalizes a data vector (data - mu) / sigma

Parameters

  • data (jax.numpy.ndarray) – A data vector or array

  • axis (int) – For nd arrays, the axis along which the data normalization will be done

Returns

A tuple comprising of:

  • Normalized data vector/array

  • Mean value(s)

  • Standard deviation value(s)

Return type

(jax.numpy.ndarray, jax.numpy.ndarray, jax.numpy.ndarray)