PyTorch手册
文章平均质量分 92
RicardoOzZ
量化投资领域的羔羊
展开
-
Pytorch 数据集太大,内存不够怎么办?——幻方量化优雅的解决方法
Tensorflow有着专门的数据读取模块tfrecord,可以直接从硬盘读取数据,而不受内存限制,但Pytorch一直没有一个专有的、高效的数据读取方法。原创 2023-12-01 16:02:01 · 1003 阅读 · 0 评论 -
Pytorch 源码学习笔记梳理
一、PyTorch的主要模块1. torch 模块torch模块包含一些激活函数、Pytorch张量的运算操作、及特定形状张量的生成。2. torch.Tensor 模块torch.Tensor模块定义了torch中张量的类型。同时,张量的类中也包含着一系列的方法,可以操作当前的张量,从而返回新的张量或更改当前的张量。torch.Storage 模块则负责torch.Tensor底层的数据存储,为张量分配连续的一维内存地址。这里需要提到一点,如果张量的某个类方法会返回张量,安装PyTorch中的原创 2021-12-10 12:56:27 · 1679 阅读 · 0 评论