Pytorch常用tensor运算之mean()求平均

本文转载自查看原文 2020-12-11 20:40 6322

dim=0,按行求平均值，返回的形状是（1，列数）

dim=1,按列求平均值，返回的形状是（行数，1）

1 x = torch.randn(2, 2, 2)
2 x

1 tensor([[[-0.7596, -0.4972],
2          [ 0.3271, -0.0415]],
3 
4         [[ 1.0684, -1.1522],
5          [ 0.5555,  0.6117]]])

1 x.mean(-3)

1 tensor([[ 0.1544, -0.8247],
2         [ 0.4413,  0.2851]])

1 x.mean(-3).shape

1 torch.Size([2, 2])

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 Matlab求平均值函数mean Pytorch-tensor的转置，运算 [Pytorch]Pytorch中tensor常用语法 Pytorch学习笔记（二）——Tensor Pytorch：Tensor 张量操作 pytorch创建tensor数据 pytorch tensor 的拼接和拆分 Pytorch之Tensor大详解 pytorch教程[2] Tensor的使用 Pytorch 扩展Tensor维度、压缩Tensor维度