pytorch

5542ff76 - repro to_channels_last also for 3d