Deform the tensor using torch.view. from __future__ import print_function. import torch. x = torch.randn(8, 8). z = x.view(-1, 4). print(x.size(), z.size()).
確定! 回上一頁