python - Convert image to proper dimension PyTorch - Stack ...
https://stackoverflow.com/questions/5065744902.06.2018 · Convert image to proper dimension PyTorch. Ask Question Asked 3 years, 7 months ago. Active 3 years, 7 months ago. Viewed 7k times 3 I have an input image, as numpy array of shape [H, W, C] where H - height, W - width and C - channels. I want to convert it into [B, C, H, W] where B - batch size, which should be equal to 1 ...