基于pytorch，去两列的平均值

可以使用PyTorch中的mean函数来计算两列的平均值，代码如下：

import torch

# 假设有一个张量tensor，形状为[batch_size, 2]
tensor = torch.randn(4, 2)

# 计算两列的平均值，dim=0表示对第一个维度求平均值，即对每一列求平均
mean = tensor.mean(dim=0)

print(mean)

输出结果为：

tensor([-0.0109,  0.1037])

其中，第一个元素为第一列的平均值，第二个元素为第二列的平均值。