torch.stack()是将原来的几个tensor按照一定方式进行堆叠,然后在按照堆叠后的维度进行切分。 ... torch.cat,torch.stack,torch.vstack,torch.hstack.
確定! 回上一頁