torch二维张量的第一列乘2，第二列乘3

在PyTorch中，你可以使用`torch`库操作张量。对于二维张量（即矩阵），如果你想要将第一列元素乘以2，第二列元素乘以3，可以这样做：

import torch

# 假设 tensor_a 是一个二维张量，例如：
tensor_a = torch.tensor([[1, 2], [3, 4], [5, 6]])

# 首先，你需要获取第一列和第二列，这可以通过索引来完成：
col_1 = tensor_a[:, 0]  # 第一列
col_2 = tensor_a[:, 1]  # 第二列

# 然后分别对两列做乘法：
col_1_scaled = col_1 * 2
col_2_scaled = col_2 * 3

# 将结果合并回原始形状，通常使用 `unsqueeze` 来添加缺失的维度：
scaled_tensor = torch.cat((col_1_scaled.unsqueeze(-1), col_2_scaled.unsqueeze(-1)), dim=1)

print(scaled_tensor)

上述代码将会得到一个新的二维张量，其中第一列是原第一列的两倍，第二列是原第二列的三倍。