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