1. torch.stack() 方法：
- 行为： 创建一个新的维度，并在该维度上堆叠张量。结果张量的维度比原始张量多一维。
- 适用场景： 当需要在新的维度上堆叠张量时，特别是在创建新批次或样本时。
import torch
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])
# 沿着新的维度（第0维）堆叠
result = torch.stack((tensor1, tensor2))
print(result)
# 输出:
# tensor([[[1, 2],
# [3, 4]],
# [[5, 6]]])
2. torch.cat() 与 torch.unsqueeze() 方法：
-
行为： 使用
torch.unsqueeze()在现有维度上增加一个维度，然后使用torch.cat()连接张量。这样可以在现有维度上进行连接，但增加了一个额外的维度。 - 适用场景： 当希望在现有维度上连接张量，但需要处理张量的不同形状时，可以使用此方法。
import torch
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])
# 使用 torch.unsqueeze 在现有维度上增加一个维度，然后使用 torch.cat() 连接
tensor2_expanded = torch.unsqueeze(tensor2, dim=0)
result = torch.cat((tensor1, tensor2_expanded), dim=0)
print(result)
# 输出:
# tensor([[1, 2],
# [3, 4],
# [5, 6]])
3. torch.cat() 与 torch.chunk() 方法：
-
行为： 使用
torch.chunk()将一个张量沿着指定维度分割成多个张量，然后通过torch.cat()连接所需的部分。这样可以在分割点处灵活地进行连接。 - 适用场景： 当需要在连接之前在指定点处对张量进行分割时，特别是处理不同部分的不同操作。
import torch
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])
# 使用 torch.chunk() 将 tensor1 沿着第0维度分割成两部分，然后通过 torch.cat() 连接所需的部分
chunks = torch.chunk(tensor1, chunks=2, dim=0)
result = torch.cat((chunks[0], tensor2, chunks[1]), dim=0)
print(result)
# 输出:
# tensor([[1, 2],
# [5, 6],
# [3, 4]])
4. torch.cat() 与 Python 列表方法：文章来源：https://www.toymoban.com/news/detail-796757.html
-
行为： 使用 Python 列表来包含张量，然后使用
torch.cat()连接列表中的张量。这样提供了一种更灵活的方式来组织和连接张量。 - 适用场景： 当有多个张量需要连接，而这些张量以列表的形式存在时，可以使用此方法。这种方法对于动态构建需要连接的张量集合很有用。
import torch
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])
# 使用 Python 列表包含张量，然后使用 torch.cat() 连接列表中的张量
tensors_list = [tensor1, tensor2]
result = torch.cat(tensors_list, dim=0)
print(result)
# 输出:
# tensor([[1, 2],
# [3, 4],
# [5, 6]])
相关链接：torch.cat()函数的理解文章来源地址https://www.toymoban.com/news/detail-796757.html
到了这里，关于PyTorch 还提供的几种连接张量的方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
