部分内容摘自pytorch中文文档torch.cat torch.cat(inputs, ...torch.cat()可以看 ... torch.cat(dim=0) = torch.vstack() torch.cat(dim=1) = torch.hstack() ...
確定! 回上一頁