normalize

fun <I, T : DType, V> Transform<I, Tensor<T, V>>.normalize(ctx: ExecutionContext, mean: FloatArray, std: FloatArray, channelAxis: Int = -1): Transform<I, Tensor<T, V>>(source)

Chains a Normalize transform that applies channel-wise normalization.

Parameters

ctx

The execution context

mean

Per-channel mean values

std

Per-channel standard deviation values

channelAxis

The axis containing channels (default: -1)