torch.flatten()函数_如切如磋-程序员宅基地. 1)flatten(x,1)是按照x的第1个维度拼接(按照列来拼接,横向拼接); 2)flatten(x,0)是按照x的第0个维度拼接(按照行来 ...
確定! 回上一頁