『pytorch』PyTorch 基础

一、基本数据：Tensor

1. Tensor的组合与分块

组合

组合操作是指将不同的 Tensor 叠加起来，主要有 torch.cat() 和 torch.stack() 两个函数。

• torch.cat()：即 concatenate，是指沿着已有的数据的某一维度进行拼接，操作后数据的总维数不变，在进行拼接时，除了拼接的维度之外，其他维度必须相同。
• torch.stack()：指新增维度，并按照指定的维度进行叠加。

分块

分块是与组合相反的操作，指将 Tensor 分割成不同的子 Tensor，主要有 torch.chunk() 和 torch.split() 两个函数。

• torch.chunk()：需要指定分块的数量
• torch.split()：需要指定每一块的大小

2. Tensor的索引与变形