OK, I’ve got desired shape,Thanks!!!
BTW, I was browsing around and I found that in this link
ByteTensor to FloatTensor is slow? - PyTorch Forums 2
Why do people always convert tensor’s type to float while calculating, would that be faster?