For example, I use numpy:

2017-12-05-170222_544x733_scrot.png544×733 41.1 KB

Then I would like to do it in PyTorch, but it failed, I do not know how to solve it.

2017-12-05-170351_1042x765_scrot.png1042×765 47 KB