pytorch中的切片时的省略号

x = torch.randint(0,10,(2,3,4))
print(x)print(x[...,:2]==x[:,:,:2])# 当x有3个维度时，这二者相等。

x[…, :2] # 前面的维度都取，最后一个维度只取前两个元素。

x = torch.randint(0,10,(2,3,4))
print(x)
print('*** 单个元素  ***')
print(x[...,1],x[...,1].shape)print('*** 多个元素  ***')
print(x[...,:2],x[...,:2].shape)

输出情况。只取单个元素时，变量的维度数减少了1个。取多个元素时，维度数不变

tensor([[[9, 1, 4, 6],[7, 0, 7, 1],[7, 1, 8, 5]],[[7, 8, 3, 9],[6, 9, 0, 2],[8, 5, 3, 1]]])
*** 单个元素  ***
tensor([[1, 0, 1],[8, 9, 5]]) torch.Size([2, 3])
*** 多个元素  ***
tensor([[[9, 1],[7, 0],[7, 1]],[[7, 8],[6, 9],[8, 5]]]) torch.Size([2, 3, 2])

pytorch中的切片时的省略号相关推荐

pyTorch中tensor运算
文章目录 PyTorch的简介 PyTorch中主要的包 PyTorch的安装使用GPU的原因使数据在GPU上运行什么使Tensor(张量) 一些术语介绍 Tensor的属性介绍(Rank,ax ...
如何用切片工具做html网页,ps中的切片工具怎么用，怎么将html文本添加到切片...
显示切片选项对话框执行下列操作之一: 使用切片选择工具双击切片.(如果该工具未显示,请按住"裁剪"或"切片"工具). 如果切片选择工具是现用的,则单击选项栏 ...
详解Pytorch中的requires_grad、叶子节点与非叶子节点、with torch.no_grad()、model.eval()、model.train()、BatchNorm层
requires_grad requires_grad意为是否需要计算梯度使用backward()函数反向传播计算梯度时,并不是计算所有tensor的梯度,只有满足下面条件的tensor的梯度才会被 ...
Pytorch中卷积层转向全连接层时，全连接层输入维度的确定
Pytorch中卷积层转向全连接层时,全连接层输入维度的确定一.代码法: 改变的AlexNet网络结构: AlexNet 二.公式法: 一.代码法: 改变的AlexNet网络结构: 输入维度为 ( ...
svg中如何让文本溢出时显示省略号
svg中如何让文本溢出时显示省略号最近在使用d3.js实现一个树状组织结构图,遇到如下图所示的问题,要求是文本内容固宽,溢出内容应用省略号代替,如果是使用普通html标签实现的话,是比较简单的,ov ...
Pytorch中Tensor的索引，切片以及花式索引(fancy indexing)
目录理解Tensor的dim 索引简单索引用1维的list,numpy,tensor索引用booltensor索引切片花式索引结语前一段时间遇到一个花式索引的问题,在搜索良久之后没有找 ...
gpu处理信号_在PyTorch中使用DistributedDataParallel进行多GPU分布式模型训练
先进的深度学习模型参数正以指数级速度增长:去年的GPT-2有大约7.5亿个参数,今年的GPT-3有1750亿个参数.虽然GPT是一个比较极端的例子但是各种SOTA模型正在推动越来越大的模型进入生产应用 ...
【深度学习】在PyTorch中使用 LSTM 进行新冠病例预测
时间序列数据,顾名思义是一种随时间变化的数据.例如,24 小时时间段内的温度,一个月内各种产品的价格,特定公司一年内的股票价格.长短期记忆网络(LSTM)等高级深度学习模型能够捕捉时间序列数据中的模式 ...
【深度学习】在PyTorch中构建高效的自定义数据集
文章来源于磐创AI,作者磐创AI 学习Dataset类的来龙去脉,使用干净的代码结构,同时最大限度地减少在训练期间管理大量数据的麻烦. 神经网络训练在数据管理上可能很难做到"大规模" ...

pytorch中的切片时的省略号

pytorch中的切片时的省略号相关推荐

最新文章

热门文章