- 博客(2)
- 收藏
- 关注
原创 detach 函数理解笔记
detach函数表示将tensor变量的梯度删除,从而实现减少内存的目的 举个简单的例子: # 自动求梯度 import torch x = torch.tensor(1.,requires_grad = True) w = torch.tensor(2.,requires_grad = True) b = torch.tensor(3.,requires_grad = True) y = w*x+b; z = y**2; z.backward(); print('w.grad:',w.grad,'
2021-10-18 09:11:42
574
原创 DataLoader 基础学习笔记
在自学词向量时,遇到了DataLoader这个知识点,最开始不太理解,第二次学习的时候渐渐明白了它的作用。记录一下,给相似的小白学习用。 1、DataLoader的作用 DataLoader 的作用是将数据集(Dataset)自动形成一个一个的Batch,以用于批处理使用。建立DataLoader需要先建立数据集。 相关包的导入 import torch import torch.utils.data as tud; 2、DataSet的建立 2.1 直接采用 torch.utils.data
2021-10-16 09:59:21
398
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人