PyTorch 中使用 torch.unsqueeze() 扩展张量维度

使用 torch.unsqueeze() 扩展张量维度：batch_x = torch.unsqueeze(batch_x, dim=1)

这行代码的功能是将 batch_x 张量的维度扩展为 (50, 1, 10000, 12)，并在第 2 维度上添加一个维度。

假设 batch_x 的原始维度为 (50, 10000, 12)，表示有 50 个样本，每个样本有 10000 个时间步和 12 个特征。使用 torch.unsqueeze() 函数可以在指定的维度上添加一个维度。

dim=1 表示在第 2 维度上添加一个维度。添加维度后，batch_x 的维度变为 (50, 1, 10000, 12)，其中第 2 维度的大小为 1，表示每个样本只有一个通道。

这样做的目的可能是为了符合某些模型或计算操作的输入要求。例如，某些卷积神经网络模型的输入要求有通道维度，因此需要在输入张量上添加一个通道维度。