torch.flatten(x)等于torch.flatten(x,0)默认将张量拉成一维的向量,也就是说从第一维开始平坦化,torch.flatten(x,1)代表 ... Pytorch:torch.flatten()与torch.nn.
確定! 回上一頁