一、基本数据:Tensor
1. Tensor的组合与分块
组合
组合操作是指将不同的 Tensor
叠加起来,主要有 torch.cat()
和 torch.stack()
两个函数。
torch.cat()
:即concatenate
,是指沿着已有的数据的某一维度进行拼接,操作后数据的总维数不变,在进行拼接时,除了拼接的维度之外,其他维度必须相同。torch.stack()
:指新增维度,并按照指定的维度进行叠加。
分块
分块是与组合相反的操作,指将 Tensor
分割成不同的子 Tensor
,主要有 torch.chunk()
和 torch.split()
两个函数。
torch.chunk()
:需要指定分块的数量torch.split()
:需要指定每一块的大小