Pytorch torch.mean()的简单用法
简单来说就是求平均数。
比如以下的三种简单情况:
import torchx1 = torch.Tensor([1, 2, 3, 4])
x2 = torch.Tensor([[1],[2],[3],[4]])
x3 = torch.Tensor([[1, 2],[3, 4]])
y1 = torch.mean(x1)
y2 = torch.mean(x2)
y3 = torch.mean(x3)
print(y1)
print(y2)
print(y3)
输出:
tensor(2.5000)
tensor(2.5000)
tensor(2.5000)
也就是说,在没有指定维度的情况下,就是对所有数进行求平均。
更多的时候用到的是有维度的情形,如:
import torchx = torch.Tensor([1, 2, 3, 4, 5, 6]).view(2, 3)
y_0 = torch.mean(x, dim=0)
y_1 = torch.mean(x, dim=1)
print(x)
print(y_0)
print(y_1)
输出:
tensor([[1., 2., 3.],[4., 5., 6.]])
tensor([2.5000, 3.5000, 4.5000])
tensor([2., 5.])
输入tensor的形状为(2, 3),其中2为第0维,3为第1维。对第0维求平均,得到的结果为形状为(1, 3)的tensor;对第1维求平均,得到的结果为形状为(2, 1)的tensor。
可以理解为,对哪一维做平均,就是将该维所有的数做平均,压扁成1层(实际上这一层就给合并掉了,比如上面的例子,2维的tensor在求平均数后变成了1维),而其他维的形状不影响。
如果要保持维度不变(例如在深度网络中),则可以加上参数keepdim=True:
y = torch.mean(x, dim=1, keepdim=True)
Pytorch torch.mean()的简单用法相关推荐
- Pytorch torch.rsqrt()的简单用法
官方文档:https://pytorch.org/docs/master/generated/torch.rsqrt.html 简单来说就是对每个元素取平方根后再取倒数,即: out i=1input ...
- Pytorch torch.topk()的简单用法
官方文档:https://pytorch.org/docs/stable/generated/torch.topk.html?highlight=topk#torch.topk 由于numpy本身是没 ...
- Pytorch torch.manual_seed()的简单用法
简单来说就是指定随机数种子(种子可以是个任意int值),用来生成伪随机数: import torchtorch.manual_seed(1) print(torch.rand(1)) print(to ...
- Pytorch torch.nonzero()的简单用法
官方文档:https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/ 顾名思义,返回非零元素的索引: 先从一维的看起: ...
- Pytorch nn.DataParallel()的简单用法
简单来说就是使用单机多卡进行训练. 一般来说我们看到的代码是这样的: net = XXXNet() net = nn.DataParallel(net) 这样就可以让模型在全部GPU上训练. 方法定义 ...
- Pytorch transforms.Resize()的简单用法
简单来说就是调整PILImage对象的尺寸,注意不能是用io.imread或者cv2.imread读取的图片,这两种方法得到的是ndarray. 将图片短边缩放至x,长宽比保持不变: transfor ...
- pytorch tensor查找0_Pytorch简单教程
2019 年,ML 框架之争只剩两个实力玩家:PyTorch 和 TensorFlow.研究者大批涌向 PyTorch,而业界的首选仍然是 TensorFlow. Pytorch和TensorFlow ...
- 如何使用PyTorch torch.max()
In this article, we'll take a look at using the PyTorch torch.max() function. 在本文中,我们将介绍如何使用PyTorch ...
- Pytorch中的collate_fn函数用法
Pytorch中的collate_fn函数用法 官方的解释: Puts each data field into a tensor with outer dimension batch size ...
最新文章
- 你不曾见过的酷炫地图可视化作品(一)
- Redux你的Angular 2应用--ngRx使用体验
- 知识图谱最新论文清单,高阶炼丹师为你逐一解读
- python大数_python处理大数字的方法
- python echo命令_如何用Python调用外部命令
- Problem B: 求各位数字之和
- 最新版的SSM框架spring5.0搭建教程(附源码)
- 每一个企业的供应商都有供应商
- 操作系统两大创始人反目,这个排名第九的 Linux 发行版 OS 何去何从?
- c 按输入的字母来输出对应效果
- OSN8800 华为单板
- Spring Boot接支付宝第三方支付(沙箱)
- 苹果手机左上角的数字怎么弄_手把手教你把抖音视频音乐设置为手机铃声。
- 以下关于python缩进的描述_以下关于Python缩进的描述中,错误的是
- 给SpringBoot Web应用配上JavaFx漂亮衣服
- VeraCrypt命令行下加载加密磁盘
- 程序化随机多边形地图生成
- Nokia Open MSS 学习个人笔记
- 华为2021届实习机试题
- python初学者爬虫教程(二)动态网页抓取
热门文章
- 浙江农林大学有计算机专业,浙江农林大学计算机科学与技术专业在职研究生
- 乌邦图怎么装php,Ubuntu如何安装php7?
- python画散点图带直线和图例_带图例的Matplotlib散点图
- vue中computed与watch的区别
- 修改windows cmd f2快捷_解放你的右手,实测12个超好用的自带快捷键
- exc导入mysql phpcms_PHP如何将EXCEL导入MYSQL,急!!!急!!哪位大师能帮帮忙啊,给个详细代码...
- 对象测试_心理测试:你会选择跟对象去吃什么夜宵?测你治愈失恋的方法是什么...
- 算法导论 思考题12-1
- 【IEEE Transactions NNLS】DSAN: Deep Subdomain Adaptation Network for Image Classification译读笔记
- .NET平台下WEB应用程序的部署(安装数据库和自动配置,启动条件)