channelsFirst

open override val channelsFirst: Boolean = false
Content copied to clipboard

If true it means that the second dimension is related to number of channels in image has short notation as NCWH, otherwise, channels are at the last position and has a short notation as NHWC.