pytorch 에서는 shape 이나 size() 를 통해 variable 또는 torch.FloatTensor 의 shape 을 확인 할 수 있다. img = Variable(img.cuda()) ...
確定! 回上一頁