torch.stack() 和 torch.cat() 都是拼接tensor常用操作,stack()可以看做并联,cat()为串联。 torch.stack() 官网:https://pytorch.org/docs/
確定! 回上一頁