Hi all, let’s say I have a 4 dims tensor:
inputs = tensor.randn(N, C, H, W)
I want to get:
inputs = [input_1, input_2, …, input_n]
input_i.shape = [C, H, W]
len(inputs) = N
Is there any elegant way to make this conversion? I can make it by a for loop but just feel not good.
This might be a weird conversion, but I do need it because the upper level of my project needs it.
Thank you for all for any suggestion!