PYTORCH中torch.mean(),dim=0,dim=1分别代表什么

举一个非常简单直观的例子：

首先，利用torch创建一个矩阵A：

A=torch.tensor([[1,2,3],[2,3,4]])

矩阵A的大小为：2*3

（1）dim=0:

B=torch.mean(A.float(),dim=0)

此时打印B为：

tensor([1.5000, 2.5000, 3.5000]) #说明dim=0时，是竖着计算均值，即按列计算；

(2) dim=1:

C=torch.mean(A.float(),dim=1)

此时打印C为：

tensor([2., 3.]) #说明dim=1时，是横向计算均值，即按行计算.

PYTORCH中torch.mean(),dim=0,dim=1分别代表什么相关推荐

Pytorch中torch.nn.Softmax的dim参数含义
自己搞了一晚上终于搞明白了,下文说的很透彻,做个记录,方便以后翻阅 Pytorch中torch.nn.Softmax的dim参数含义
torch.cat 关于 dim=0,dim=1 测试
文章目录 1.torch.cat 2. 代码 3. 结果 4. 总结: 1.torch.cat 官方说明:torch.cat torch.cat(tensors, dim=0, *, out=None ...
pytorch中torch.optim的介绍
pytorch中torch.optim的介绍这是torch自带的一个优化器,里面自带了求导,更新等操作.开门见山直接讲怎么使用: 常用的引入: import torch.optim as optim ...
Pytorch中torch.argmax()函数解析
一. torch.argmax()函数解析 1. 官网链接 torch.argmax(),如下图所示: 2. torch.argmax(input)函数解析 torch.argmax(input) → ...
PyTorch 中 torch.optim优化器的使用
一.优化器基本使用方法建立优化器实例循环: 清空梯度向前传播计算Loss 反向传播更新参数示例: from torch import optim input = ..... optimiz ...
【PyTorch】torch.mean(), dim=0, dim=1
对于torch.mean(s,dim=0),表示跨行求平均. 同理,对于dim=1为第一个维度,代表列. 对于torch.sum(s,dim=1),表示跨列求平均.
pytorch中x.norm(p=2,dim=1,keepdim=True)的理解
代码:x.norm(p=2,dim=1,keepdim=True) 功能:求指定维度上的范数. 函数原型:[返回输入张量给定维dim 上每行的p范数] torch.norm(input, p, dim ...
torch.sum(),dim=0,dim=1解析
直接上代码: # 定义一个2维张量a a = torch.tensor([[1,2,3],[4,5,6]]) print(a) a.shape output: tensor([[ 1, 2, 3],[ ...
PyTorch中torch.norm函数详解
torch.norm() 是 PyTorch 中的一个函数,用于计算输入张量沿指定维度的范数.具体而言,当给定一个输入张量 x 和一个整数 p 时,torch.norm(x, p) 将返回输入张量 x ...
Pytorch中torch.unsqueeze()和torch.squeeze()函数解析
一. torch.squeeze()函数解析 1. 官网链接 torch.squeeze(),如下图所示: 2. torch.squeeze()函数解析 torch.squeeze(input, di ...

PYTORCH中torch.mean(),dim=0,dim=1分别代表什么

PYTORCH中torch.mean(),dim=0,dim=1分别代表什么相关推荐

最新文章

热门文章