Pytorch深度学习随手记（1）x = torch.squeeze(x)

最新推荐文章于 2023-11-10 20:19:57 发布
最新推荐文章于 2023-11-10 20:19:57 发布
阅读量503 收藏
点赞数
分类专栏： 从入门到放弃 文章标签： pytorch 深度学习 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Promise11161314/article/details/121082479
版权

squeeze的用法：对数据的维度进行压缩或者解压

torch.squeeze() 对数据的维度进行压缩

        squeeze(a)就是将a中所有为1的维度删掉。

        a.squeeze(N) 就是去掉a中指定的维数为N的维度。

        b=torch.squeeze(a，N) ：b=a中去掉指定的N维的数据。

torch.unsqueeze()对数据维度进行扩充

        a.unsqueeze(N) 就是在a中指定位置N加上一个维数为1的维度。

        b=torch.unsqueeze(a，N) b=a中指定位置N加上一个维数为1的数据

演示：

 

为了强大自身而学习
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
详解pytorch中squeeze()和unsqueeze()函数介绍
12-17
squeeze的用法主要就是对数据的维度进行压缩或者解压。 先看torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.squeeze(a，N) a中去掉指定的定的维数为一的维度。 再看torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度，比如原本有个三行的数据（3
Pytorch操作笔记
weixin_43341824的博客
02-23 91
torch.Tensor.repeat(*sizes) 沿着指定的维度重复tensor。 不同于’expand()’，本函数复制的是tensor中的数据 参数 -*sizes(torch.Size ot int…) 沿着每一维重复的次数 >>> x = torch.Tensor([1, 2, 3]) >>> x.repeat(4, 2) #沿着第一维重...
torch函数的学习笔记
一个小男孩
05-02 482
【莫烦Python】PyTorch 神经网络_哔哩哔哩_bilibilihttps://www.bilibili.com/video/BV1Vx411j7kT?p=7 .ToTensor()： transf = transforms.ToTensor() img_tensor = transf(img) .Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))： 均值和标准差 之前版本导入激活函数： import torch.nn.functional as
PyTorch中torch.squeeze() 和torch.unsqueeze()用法
艾文
12-09 990
squeeze的用法主要就是对数据的维度进行压缩或者解压 torch.squeeze() 对数据的维度进行压缩 https://pytorch.org/docs/stable/torch.html?highlight=torch%20squeeze#torch.squeeze (a) 去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3...
吴恩达深度学习之np.squeeze()
Fred_18的博客
06-19 884
吴恩达深度学习课程参考代码：链接：https://pan.baidu.com/s/169hDtOztbAXwnCVBhG9nug 提取码：6r52 最近学习该课程遇到的大坑，在课后作业/代码作业/第二课第一周编程作业/assignment1/2.Regularization_original.ipynb中有一段导入数据代码： train_X, train_Y, test_X, test_Y = l...
torch unsqueeze()详解
weixin_45625942的博客
11-16 9452
pytorch unsqueeze函数总结
python中torch.unsqueeze()函数理解
最新发布
AI_dataloads的博客
11-10 2166
是PyTorch库中的一个函数，它被用来在输入张量的特定维度上增加一个维度。这是一个很常用的函数，可以用来改变张量的形状和结构。函数的参数是输入张量和要增加的维度的位置。具体来说，如果你有一个形状为(N, M)的二维张量，会在第一维（即N）后面增加一个维度，得到形状为(1, N, M)的张量。如果你在第二维（即M）后面增加一个维度，那么你会得到形状为(N, 1, M)的张量。
随笔记录——torch.squeeze/torch.unsqueeze
XD的博客
03-08 166
使用squeeze()移除输入的tensor中，大小为1的维度 import torch x = torch.zeros(2, 1, 2, 1, 2) y = torch.squeeze(x) y.shape torch.Size([2, 2, 2]) 使用squeeze()移除输入的tensor中，指定维度的大小为1的维度 y = torch.squeeze(x, 0) y.shape torch.Size([2, 1, 2, 1, 2]) y = torch.squeeze(x, 1) y.sha
torch中squeeze() 和unsqueeze() 的作用
这个年纪的_我
07-18 2155
torch.squeeze torch.squeeze(input, dim=None, out=None) 将输入张量形状中的1去除并返回。 如果输入是形如(A×1×B×1×C×1×D)，那么输出形状就为：(A×B×C×D) 当给定dim时，那么挤压操作只在给定维度上。例如，输入形状为:(A×1×B),squeeze(input, 0)将会保持张量不变，只有用squeeze(input, 1)，形状会变成(A×B)。 注意： 返回张量与输入张量共享内存，所以改变其中一个的内容会改...
4.10 高级神经网络结构-DQN 强化学习
敲代码的乔帮主
04-09 3599
目录 1.写在前面 2.模块导入和参数设置 3.神经网络 4.DQN体系 5.训练 6.完整代码演示 7.附加A3C 1.写在前面 Torch 是神经网络库, 那么也可以拿来做强化学习,你同样也可以用 PyTorch 来实现, 这次我们就举 DQN 的例子, 我对比了我的 Tensorflow DQN 的代码, 发现 PyTorch 写的要简单很多. 2.模块...
pyTorch.randn()、rand()、cat()、pow()、scatter_()、.squeeze() 、.unsqueeze()
01-20
文章目录torch.randn()torch.rand()torch.cat()torch.pow().item().scatter_().squeeze().unsqueeze() torch.randn() torch.randn(*sizes, out=None) → Tensor 返回一个张量，包含了从标准正态分布(均值为0，方差为...
pytorch实现onehot编码转为普通label标签
01-20
label = torch.topk(one_hot, 1)[1].squeeze(1) print(label) tensor([[0, 0, 1], [0, 1, 0], [0, 1, 0]]) tensor([2, 1, 1]) 以上这篇pytorch实现onehot编码转为普通label标签就是小编分享给大家的全部内
squeezenet-torch:将SqueezeNet移植到PyTorch中；主要是为了我自己的学习
03-21
挤压火炬 将SqueezeNet移植到PyTorch中；主要是为了我自己的学习
pytorch 加载(.pth)格式的模型实例
12-31
有一些非常流行的网络如 resnet、squeezenet、densenet等在pytorch里面都有，包括网络结构和训练好的模型。 pytorch自带模型网址：https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-models/ 按...
【深度学习】torch.squeeze()移除维度函数 | torch.unsqueeze()增加某一维度函数 | pytorch
weixin_46274756的博客
11-29 3090
【深度学习】torch.squeeze()移除维度函数 | torch.unsqueeze()增加某一维度函数 | pytorch
函数伪代码_DQN PyTorch代码详解
weixin_32916385的博客
01-09 2396
1 简介本文参考莫烦Python。由于莫烦老师在视频中只是大致介绍了DQN的代码结构，没有对一些细节进行讲解。因此，本文基于莫烦老师的代码，针对代码的每一行进行了解释。2 相关资料网址01 《什么是DQN》什么是 DQN - PyTorch | 莫烦Python​mofanpy.com02 《DQN强化学习》DQN 强化学习 - PyTorch | 莫烦Python​mofanpy.com03 《...
莫烦python讲得好差_莫烦PYTHON——PyTorch——DQN 代码详解
weixin_39912163的博客
11-26 1653
import torch # 导入torchimport torch.nn as nn # 导入torch.nnimport torch.nn.functional as F # 导入torch.nn.functionalimport numpy...
【深度学习笔记】torch.unsqueeze()
学习带来的快乐最持久~
09-07 472
一、作用 主要是对数据维度进行扩充。给指定位置加上维数为1的维度，比如原本有个4行的数据（3），在0的位置加了一维就变成1行4列（1,3）。a.unsqueeze(N) 就是在a中指定位置N加上一个维数为1的维度。还有一种形式就是b=torch.unsqueeze(a，N) a就是在a中指定位置N加上一个维数为1的维度。 二、例子 x = torch.tensor([1, 2, 3, 4]) torch.unsqueeze(x, 0).size() 结果：torch.Size([1, 4]) torch
Pytorch中 torch.squeeze() 和torch.unsqueeze()的相关用法
荷叶田田
01-08 8903
#unsqueeze:扩充数据维度，在0起的指定位置N加上维数为1的维度 #squeeze:   维度压缩，在0起的指定位置N，去掉维数为1的的维度 torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度，比如原本有个三行的数据（3），在0的位置加了一维就变成一行三列（1,3）。a.squeeze(N) 就是在a中指定位置N加上一个维数为1的维度。还有一...
def choose_action(self, state): state = np.array([state]) # 先转成数组再转tensor更高效 state = torch.tensor(state, dtype=torch.float).to(self.device) dist = self.actor(state) value = self.critic(state) action = dist.sample() probs = torch.squeeze(dist.log_prob(action)).item() if self.continuous: action = torch.tanh(action) else: action = torch.squeeze(action).item() value = torch.squeeze(value).item() return action, probs, value
06-11
这段代码是一个 PyTorch 实现的 Actor-Critic 算法中的选择动作函数，它接收一个状态 state，并输出一个动作 action、一个概率 probs 和一个价值 value。 具体来说，这个函数首先将 state 转换成一个 numpy 数组，...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值