torch数据加载
"""
批训练,把数据变成一小批一小批数据进行训练。
DataLoader就是用来包装所使用的数据,每次抛出一批数据
"""
import torch
import torch.utils.data as Data
BATCH_SIZE = 3
x = torch.linspace(1, 10, 10) # linspace: 返回一个1维张量,包含在区间start和end上均匀间隔的step个点
y = torch.linspace(10, 1, 10)
把数据放在数据集中
torch_dataset = Data.TensorDataset(x, y)
print(torch_dataset[1])
loader = Data.DataLoader(
# 从数据集中每次抽出batch size个样本
dataset=torch_dataset,
batch_size=BATCH_SIZE,
shuffle=True,
num_workers=3,
)
def show_batch():
for epoch in range(3): # epoch: 迭代次数
print('Epoch:', epoch)
for batch_id, (batch_x, batch_y) in enumerate(loader):
print(" batch_id:{}, batch_x:{}, batch_y:{}".format(batch_id, batch_x, batch_y))
# print(f' batch_id:{batch_id}, batch_x:{batch_x}, batch_y:{batch_y}')
if name == 'main':
show_batch()
(tensor(2.), tensor(9.))
Epoch: 0
batch_id:0, batch_x:tensor([4., 5., 3.]), batch_y:tensor([7., 6., 8.])
batch_id:1, batch_x:tensor([1., 8., 9.]), batch_y:tensor([10., 3., 2.])
batch_id:2, batch_x:tensor([ 2., 7., 10.]), batch_y:tensor([9., 4., 1.])
batch_id:3, batch_x:tensor([6.]), batch_y:tensor([5.])
Epoch: 1
batch_id:0, batch_x:tensor([2., 6., 4.]), batch_y:tensor([9., 5., 7.])
batch_id:1, batch_x:tensor([9., 8., 3.]), batch_y:tensor([2., 3., 8.])
batch_id:2, batch_x:tensor([ 1., 7., 10.]), batch_y:tensor([10., 4., 1.])
batch_id:3, batch_x:tensor([5.]), batch_y:tensor([6.])
Epoch: 2
batch_id:0, batch_x:tensor([2., 7., 6.]), batch_y:tensor([9., 4., 5.])
batch_id:1, batch_x:tensor([ 3., 10., 4.]), batch_y:tensor([8., 1., 7.])
batch_id:2, batch_x:tensor([5., 1., 8.]), batch_y:tensor([ 6., 10., 3.])
batch_id:3, batch_x:tensor([9.]), batch_y:tensor([2.])
原文地址:https://www.cnblogs.com/wana-/p/15394352.html
- 易企秀前端压缩源码分析与还原
- 关于 JS 拖拽功能的冲突问题及解决方法
- 使用 SVG 和 JS 创建一个由星形变心形的动画
- 如何使用 Bootstrap 搭建更合理的 HTML 结构
- Java遍历Map对象的四种方式
- java 线程public void run()中值如何返回
- Jackson与spring框架整合的坑
- spring shiro整合时自动注入的问题
- mybatis获取update的id
- 点击!AWD攻防解题技巧在此!
- Docker 基础技术之 Linux namespace 详解
- Centos6下使用yum安装MariaDB
- Linux下部署Samba服务环境的操作记录
- SCP和Rsync远程拷贝的几个技巧
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法