百度360必应搜狗淘宝本站头条
vue dispatchspring gatewayorder by ascidea eval resetupdate canceled
当前位置：网站首页 > 技术分类 > 正文

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

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

切片

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

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

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



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

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)


相关推荐

Jquery 详细用法

1、jQuery介绍(1)jQuery是什么?是一个js框架，其主要思想是利用jQuery提供的选择器查找要操作的节点,然后将找到的节点封装成一个jQuery对象。封装成jQuery对象的目的有...

前端开发79条知识点汇总

1.css禁用鼠标事件2.get/post的理解和他们之间的区别http超文本传输协议（HTTP）的设计目的是保证客户机与服务器之间的通信。HTTP的工作方式是客户机与服务器之间的请求-应答协议。...

js基础面试题92-130道题目

92.说说你对作用域链的理解参考答案：作用域链的作用是保证执行环境里有权访问的变量和函数是有序的，作用域链的变量只能向上访问，变量访问到window对象即被终止，作用域链向下访问变量是不被允许的。...

Web前端必备基础知识点，百万网友:牛逼

1、Web中的常见攻击方式1.SQL注入------常见的安全性问题。解决方案：前端页面需要校验用户的输入数据（限制用户输入的类型、范围、格式、长度），不能只靠后端去校验用户数据。一来可以提高后端处理...

事件——《JS高级程序设计》

一、事件流1.事件流描述的是从页面中接收事件的顺序2.事件冒泡（eventbubble）：事件从开始时由最具体的元素（就是嵌套最深的那个节点）开始，逐级向上传播到较为不具体的节点（就是Docu...

前端开发中79条不可忽视的知识点汇总

过往一些不足的地方，通过博客，好好总结一下。1.css禁用鼠标事件...

Chrome 开发工具之Network

经常会听到比如"为什么我的js代码没执行啊？","我明明发送了请求，为什么反应？","我这个网站怎么加载的这么慢？"这类的问题，那么问题既然存在，就需要去解决它，需要解决它，首先我们得找对导致问题的原...

轻量级 React.js 虚拟美化滚动条组件RScroll

前几天有给大家分享一个Vue自定义滚动条组件VScroll。今天再分享一个最新开发的ReactPC端模拟滚动条组件RScroll。...

一文解读JavaScript事件对象和表单对象

前言相信做网站对JavaScript再熟悉不过了，它是一门脚本语言，不同于Python的是，它是一门浏览器脚本语言，而Python则是服务器脚本语言，我们不光要会Python，还要会JavaScrip...

Python函数参数黑科技:*args与**kwargs深度解析

90%的Python程序员不知道，可变参数设计竟能决定函数的灵活性和扩展性！掌握这些技巧，让你的函数适应任何场景！一、函数参数设计的三大进阶技巧...

深入理解Python3密码学:详解PyCrypto库加密、解密与数字签名

在现代计算领域，信息安全逐渐成为焦点话题。密码学，作为信息保护的关键技术之一，允许我们加密（保密）和解密（解密）数据。...

阿里Nacos惊爆安全漏洞，火速升级!(附修复建议)

前言好，我是threedr3am，我发现nacos最新版本1.4.1对于User-Agent绕过安全漏洞的serverIdentitykey-value修复机制，依然存在绕过问题，在nacos开启了...

Python模块:zoneinfo时区支持详解

一、知识导图二、知识讲解（一）zoneinfo模块概述...

Golang开发的一些注意事项(一)

1.channel关闭后读的问题当channel关闭之后再去读取它，虽然不会引发panic，但会直接得到零值，而且ok的值为false。packagemainimport"...

Python鼠标与键盘自动化指南:从入门到进阶——键盘篇

`pynput`是一个用于控制和监控鼠标和键盘的Python库...

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

一周热门
最近发表
标签列表
赣ICP备2023011147号-29