torch中view函数的用法

import torch import numpy x = torch.tensor(2,2,2) t = x.view(1,8)#输出数组的大小为1*8 t = x.view(-1,4)#输出数组的大小为2*4, -1表示自行判断 t = x.view(8)#输出数组的大小为8*1 t = x.view(-1)#输出数组的大小为1*8

>>阅读原文<<