return x, y
return x
plist = [ndim-1] + list(range(0,ndim-1))
x = x.permute(*plist)
if y is not None:
y = y.permute(*plist)
return x, y
return x
CHW = ChannelsFirst
CDHW = ChannelsFirst
After Change
for idx, _input in enumerate(inputs):
_input = _input.permute(*plist)
outputs.append(_input)
return outputs if idx > 1 else outputs[0]
CHW = ChannelsFirst
CDHW = ChannelsFirst