20250110_ PyTorch中的张量操作

---

前言

---

1、torch.cat 函数

torch.cat 函数将两个张量拼接起来，具体地是在第三个维度（dim=2）上进行拼接。注：dim取值范围是0~2

node_xy_demand = torch.cat((node_xy, node_demand[:, :, None]), dim=2)

其中所用参数为：

node_xy = reset_state.node_xy
# shape: (batch, problem, 2)
node_demand = reset_state.node_demand
# shape: (batch, problem)

若要拼接node_xy 与node_demand 需要将node_demand 进行维度拓展即 node_demand[:, :, None])

node_xy = torch.tensor([[[1, 2], [3, 4]], 
						[[<