일반 이미지(height x widht x channel)에서 torch (channel x height x width)로 변환

xs   # (64, 64, 3)
np.transpose(xs, (2, 0, 1))  # (3, 64, 64)

이미지 보이기

viz.image(o, win='name', opts=dict(width=256, height=256))