pytorch torch.utils.data.Dataset
应用
from torch.utils.data import DataLoader, Dataset
import torchclass TensorDataset(Dataset):# TensorDataset继承Dataset, 重载了__init__, __getitem__, __len__def __init__(self, data_tensor, target_tensor):self.data_tensor = data_tensorself.target_tensor = target_tensordef __getitem__(self, index):return self.data_tensor[index], self.target_tensor[index]def __len__(self):return self.data_tensor.size(0)# 生成数据
data_tensor = torch.randn(3, 2)
target_tensor = torch.rand(3)
# 将数据封装成Dataset
tensor_dataset = TensorDataset(data_tensor, target_tensor)
# 可使用索引调用数据
tensor_dataset[0] # 等同于(data_tensor[0],target_tensor[0])
# 可返回数据len
len(tensor_dataset)
API
An abstract class representing a Dataset.
All datasets that represent a map from keys to data samples should subclass it. All subclasses should overwrite getitem(), supporting fetching a data sample for a given key. Subclasses could also optionally overwrite len(), which is expected to return the size of the dataset by many Sampler implementations and the default options of DataLoader.
参考:
https://zhuanlan.zhihu.com/p/28200166
https://pytorch.org/docs/master/data.html
pytorch torch.utils.data.Dataset相关推荐
- 【PyTorch】torch.utils.data.Dataset 介绍与实战
文章目录 一.前言 二.torch.utils.data.Dataset 是什么 1. 干什么用的? 2. 长什么样子? 三.通过继承 torch.utils.data.Dataset 定义自己的数据 ...
- PyTorch—torch.utils.data.DataLoader 数据加载类
文章目录 DataLoader(object)类: _DataLoaderIter(object)类 __next__函数 pin_memory_batch() _get_batch函数 _proce ...
- 5 torch.utils.data (Dataset,TensorDataset,DataLoader)
文章目录 一.DataLoader(数据预处理) 1.DataLoader :(构建可迭代的数据装载器) 2.输出:DataLoader 的输出包含:数据和标签 二.TensorDataset(数据预 ...
- 阅读源码-理解torch.utils.data、torch.utils.data.Dataset、torch.utils.data.DataLoader的工作方式
文章目录 目标 Dataset DataLoader 应用 Dataset DataLoader 测试 知识点 Python splitlines()方法 python filter()函数 暂时先写 ...
- pytorch torch.utils.data.TensorDataset
应用 import torch import torch.utils.data as Datax = torch.linspace(1, 10, 10) y = torch.linspace(10, ...
- pytorch源码解析2——数据处理torch.utils.data
迭代器 理解 Python 的迭代器是解读 PyTorch 中 torch.utils.data 模块的关键. 在 Dataset, Sampler 和 DataLoader 这三个类中都会用到 py ...
- Pytorch 中的数据类型 torch.utils.data.DataLoader 参数详解
DataLoader是PyTorch中的一种数据类型,它定义了如何读取数据方式.详情也可参考本博主的另一篇关于torch.utils.data.DataLoader(https://blog.csdn ...
- PyTorch 源码解读之 torch.utils.data:解析数据处理全流程
目录 0 前言 1 Dataset 1.1 Map-style dataset 1.2 Iterable-style dataset 1.3 其他 dataset 2 Sampler 3 DataLo ...
- PyTorch 1.0 中文文档:torch.utils.data
译者:BXuan694 class torch.utils.data.Dataset 表示数据集的抽象类. 所有用到的数据集都必须是其子类.这些子类都必须重写以下方法:__len__:定义了数据集的规 ...
最新文章
- 【权游相关】龙的简史 | 混乱博物馆
- linux arcgis10.4安装教程,ArcGIS 10.1 for Server安装教程系列—— Linux下的单机安装
- PrintWriter write与println方法的区别
- 一些能提高ABAP开发人员日常工作效率的ABAP小工具
- SQL Server 中索引的禁用与删除
- 栈溢出笔记1.8 字符串问题
- onClientClick 和 onClient 区别
- java中文件如何加密压缩?
- Linux进程间通信——使用消息队列
- 一个单独的UDP数据包能够放入数据的最大长度是65508
- Centos7重置Mysql 8.0.1 root 密码
- SQL 常用语句INSERT INTO,UPDATE,DELETE
- 软件程序设计中的N-S图、PAD图、程序流程图、E-R图
- 高斯-勒让德积分学习
- 峰度的意义_李德荃关于偏度与峰度的讲解
- macos可以升级到指定版本吗_如何升级mac系统版本?
- 记录h5文件数据转为npy和csv格式类型的方法
- html5分镜头脚本范例,分镜头脚本教程图解
- 数据科学分布——二项式分布
- SpringCloud Gateway网关统一聚合Swagger接口文档(knife4j),实现通过网关统一文档地址查看所有子服务的接口文档
热门文章
- struts2上传文件
- CSS教程:实例讲解定位Position
- Stackint stack = new Stackint报错原因
- html尾部代码_3分钟短文:Laravel Form,让你不再写 HTML 的好“库”
- 计算机网络技术之城域网与三网融合技术
- listwidget设置条目文本居中_将Word文档中的所有图片一键居中对齐,这个方法比插件更好用!...
- php 脚本内存耗尽,php - 内存耗尽(最小脚本) - SO中文参考 - www.soinside.com
- redis远程连接不上_redis连接认证及bind的注意事项
- sql server代理无法启动_谁说前端不需要懂Nginx反向代理与负载均衡
- 手写 Promise.race