torch.empty()造成的内存溢出问题

torch.empty()造成的内存溢出问题

问题

代码实现的时候遇见了一个很奇怪的问题,运行同一段代码会出现一些奇怪的超大指数值,调试发现是混淆了torch.empty 和torch.zeros的概念。对torch.empty初始化的矩阵直接"+=",出现的内存问题。

例如:

python 复制代码
import torch

a = torch.empty((3,4))
a += 1

上述代码不会报错,但是结果不正确。

原因

这是由于torch.empty 创建了矩阵会分配一个内存地址,该地址对应的数据是随机的,可能跟你上一次的缓存或者其他数据有关

例如

python 复制代码
a = torch.empty((3,4))
#
# tensor([[6.8943e+34, 1.6212e-19, 1.4586e-19, 7.7179e+28],
#        [1.6217e-19, 1.4586e-19, 1.8617e+25, 1.6212e-19],
#        [1.4587e-19, 1.1578e+27, 3.3248e-06, 6.0542e+22]])

这里的矩阵值是随机的,因此torch.empty只起到一个占位的作用,只有需要赋值的时候可以使用,

但是如果明确需要初始值为0,需要torch.zeros

PS: 不排除有一定的机率初始化为0,这就造成程序有一些偶然机会会正常运行

相关推荐
Q_Q5110082854 分钟前
python的婚纱影楼管理系统
开发语言·spring boot·python·django·flask·node.js·php
JNU freshman5 分钟前
计算机视觉 之 经典模型汇总
人工智能·计算机视觉
苏苏susuus9 分钟前
NLP:RNN文本生成案例分享
人工智能·rnn·自然语言处理
若兰幽竹27 分钟前
【从零开始编写数据库:基于Python语言实现数据库ToyDB的ACID特性】
数据库·python
东方佑28 分钟前
仅27M参数!SamOutVX轻量级语言模型刷新认知,小身材也有大智慧
人工智能·语言模型·自然语言处理
东临碣石8230 分钟前
【AI论文】OmniPart:基于语义解耦与结构连贯性的部件感知三维生成
人工智能
xiaocainiao88135 分钟前
Python 实战:构建 Git 自动化助手
git·python·自动化
啾啾Fun1 小时前
咨询导览,AI发展趋势
人工智能
向阳逐梦1 小时前
PID控制算法理论学习基础——单级PID控制
人工智能·算法
nightunderblackcat1 小时前
新手向:使用Python将多种图像格式统一转换为JPG
开发语言·python