torch.flip(input,dim):第一个参数是输入,第二个参数是输入的第几维度,按照维度对输入进行翻转import torchx = torch.arange(16).view(2, 2, ...
確定! 回上一頁