PyTorch基础（九）----- torch.mean()方法

一、方法详解

含义：顾名思义，返回一个Tensor的均值

torch.mean(input, dim, keepdim=False)

input：输入的张量
dim：求均值的维度，若dim = 0，则沿行求均值；若dim=1，则沿列求均值
keepdim：输出张量是否跟输入张量的另一个维度相同

这个用言语很难说明白，我们直接通过案例来理解掌握！

二、案例

案例1

import torch
x = torch.randn(4, 4)
print(x.size())
y = torch.mean(x, dim=0, keepdim=True)
print(y.size())

分析

如果我们要沿dim=0这个维度相同，而且keepdim=True，那么输出张量的另一个维度dim=1就跟输入张量相同。

运行结果

案例2

import torch
x = torch.randn(4, 4)
print(x.size())
y = torch.mean(x, dim=1, keepdim=True)
print(y.size())

运行结果

PyTorch基础（九）----- torch.mean()方法相关推荐

pytorch基础-使用 TORCH.AUTOGRAD 进行自动微分（5）
在训练神经网络时,最常用的算法是反向传播.PyTorch的反向传播(即tensor.backward())是通过autograd包来实现的,autograd包会根据tensor进行过的数学运算来自动计 ...
3.Pytorch基础模块torch的API之Indexing,Slicing,Joining,Mutating Ops实例详解
文章目录 0. torch 1. Tensors 2. Creation Ops 3. Indexing,Slicing,Joining,Mutating Ops 3.1 torch.cat() 3. ...
PyTorch基础（15）-- torch.flatten()方法
前言最近在复现论文中一个块的时候需要使用到torch.flatten()这个方法,这个方法其实很简单,但其中有一些细节可能需要注意,且有个关键点很容易忘记,故在此记录以备查阅. 方法解析 flatt ...
PyTorch基础（十）----- torch.max()方法
一.前言这个方法跟上一篇文章的torch.max()方法非常类似,只不过一个是求最大值,一个是求平均值.在某些情况下,甚至可以代替下采样中的最大池化和平均池化,所以说,这两个方法的用处还是蛮大的. ...
PyTorch基础（六）----- torch.eq()方法
一.torch.eq()方法详解对两个张量Tensor进行逐元素的比较,若相同位置的两个元素相同,则返回True:若不同,返回False. torch.eq(input, other, *, out ...
深度学习之Pytorch基础教程！
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:李祖贤,Datawhale高校群成员,深圳大学随着深度学习的发展 ...
【深度学习】深度学习之Pytorch基础教程！
作者:李祖贤,Datawhale高校群成员,深圳大学随着深度学习的发展,深度学习框架开始大量的出现.尤其是近两年,Google.Facebook.Microsoft等巨头都围绕深度学习重点投资了一系 ...
PyTorch基础(part3)
学习笔记,仅供参考,有错必纠文章目录 PyTorch 基础线性回归常用代码导包生成数据构建神经网络模型非线性回归生成数据构建神经网络模型 PyTorch 基础线性回归常用代码 # ...
深度学习导论（3）PyTorch基础
深度学习导论(3)PyTorch基础一. Tensor-Pytorch基础数据结构二. Tensor索引及操作 1. Tensor索引类型 2. Tensor基础操作 3. 数值类型 4. 数值类 ...
pytorch基础知识+构建LeNet对Cifar10进行训练+PyTorch-OpCounter统计模型大小和参数量+模型存储与调用
整个环境的配置请参考我另一篇博客.ubuntu安装python3.5+pycharm+anaconda+opencv+docker+nvidia-docker+tensorflow+pytorch+C ...

PyTorch基础（九）----- torch.mean()方法

一、方法详解

input：输入的张量

dim：求均值的维度，若dim = 0，则沿行求均值；若dim=1，则沿列求均值

keepdim：输出张量是否跟输入张量的另一个维度相同

二、案例

案例1

分析

运行结果

案例2

运行结果

PyTorch基础（九）----- torch.mean()方法相关推荐

最新文章

热门文章