Support `channel_first` format

[innat]

Short Description

Currently it supports only channel last format, i.e. (bs, depth, height, width, channel). To make it compatible with torch ecosystem, it would be convenient to support channel last format also, i.e. (bs, channel, depth, height, width).