1 前言
在这篇介绍中【Pytorch】张量的维度/轴/dim的理解已经解释了dim的用法,下面具体看torch.mean()的用例。
2 一维二维
# 二维
a = torch.arange(6) * 1.
b = a.reshape(2, 3)
print(b
这篇博客详细介绍了PyTorch中torch.mean()函数的使用,通过具体的二维张量示例展示了如何计算平均值,包括沿不同维度求平均,并给出了运算结果及其形状。
在这篇介绍中【Pytorch】张量的维度/轴/dim的理解已经解释了dim的用法,下面具体看torch.mean()的用例。
# 二维
a = torch.arange(6) * 1.
b = a.reshape(2, 3)
print(b
9124
2万+

被折叠的 条评论
为什么被折叠?