pytorch的切片操作和原地赋值（pytorch切片选取元素）

ztj100 2024-11-14 19:24 11 浏览 0 评论

切片

与python的数组操作类似，pytorch的张量也支持切片操作。

构建一个4行5列的二维数组。

Bash

import torch
x = torch.arange(20).reshape(4,5)

通过切片取出前2行的数据。

Bash

x[0:2]

也可以通过切片操作给张量赋值。

给第一行的数据赋值为0

x[0:1,:]=0

原地赋值

张量在运算的过程中会申请新的内存地址，即使是同一个张量。

先来看个例子。

先构建2个张量x,y

x=torch.tensor([1])
y=torch.tensor([2])

使用函数查看张量x的内存地址为：2182261278336

将x与y相加，重新赋值给x，再查看下x的内存地址：2182257581408

可以看到前后2次查询的内存地址是不一样的，在运算过程中申请了新的内存空间，这会导致运算开销变大。

可以使用切片操作，使用原有的内存空间。

x=torch.tensor([1])
y=torch.tensor([2])
id(x)
x[:]=x+y
id(x)

再次查看运算前后的内存地址，这次就一样了，说明没有申请新的内存空间。

除了使用切片减少内存空间也可以使用+=操作。

x=torch.tensor([1])
y=torch.tensor([2])
id(x)
x+=y
id(x)

torch.arange

上一篇：PyTorch编程常规套路(上)（pytorch 60分钟教程）
下一篇：MMDetection-RetinaNet(BBox)（mmdetection bbox_overlaps）

pytorch的切片操作和原地赋值（pytorch切片选取元素）

切片

原地赋值

相关推荐

取消回复欢迎你发表评论:

Vue自定义Hook示例:useUrlState（vue中的自定义指令如何使用）

Vue-实现自定义插件弹窗（vue 实现弹窗）

旗舰机新标杆 OPPO Find X2系列正式发布售价5499元起

什么是幂等?分布式锁如何实现业务幂等?

【Python机器学习系列】建立多层感知机模型预测心脏疾病

如何发个 npm 包?

详解MySQL 字符串拼接之concat\concat_ws\group_concat

手把手教你搞定菜单权限设计，精确到按钮级别，建议收藏

如何快速切换node版本?利用n包快速切换nodejs版本

让Jenkins自动部署你的Vue项目「实践」

pytorch的切片操作和原地赋值（pytorch切片选取元素）

切片

原地赋值

微信扫一扫：分享

相关推荐

取消回复欢迎 你 发表评论:

Vue自定义Hook示例:useUrlState（vue中的自定义指令如何使用）

Vue-实现自定义插件弹窗（vue 实现弹窗）

旗舰机新标杆 OPPO Find X2系列正式发布 售价5499元起

什么是幂等?分布式锁如何实现业务幂等?

【Python机器学习系列】建立多层感知机模型预测心脏疾病

如何发个 npm 包?

详解MySQL 字符串拼接之concat\concat_ws\group_concat

手把手教你搞定菜单权限设计，精确到按钮级别，建议收藏

如何快速切换node版本?利用n包快速切换nodejs版本

让Jenkins自动部署你的Vue项目「实践」

取消回复欢迎你发表评论:

旗舰机新标杆 OPPO Find X2系列正式发布售价5499元起