Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenDocCN
pytorch-doc-zh
提交
f11de918
P
pytorch-doc-zh
项目概览
OpenDocCN
/
pytorch-doc-zh
通知
122
Star
3932
Fork
992
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
pytorch-doc-zh
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
f11de918
编写于
3月 04, 2019
作者:
飞
飞龙
提交者:
GitHub
3月 04, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update data.md
上级
99b6d8cc
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
9 addition
and
7 deletion
+9
-7
docs/1.0/data.md
docs/1.0/data.md
+9
-7
未找到文件。
docs/1.0/data.md
浏览文件 @
f11de918
# torch.utils.data
> 译者:[BXuan694](https://github.com/BXuan694)
```
py
class
torch
.
utils
.
data
.
Dataset
```
表示数据集的抽象类。
所有用到的数据集都必须是其子类。这些子类都
必须重写以下方法:
`__len__`
:定义了数据集的规模;
`__getitem__`
:支持0到len(self)范围内的整数索引。
所有用到的数据集都必须是其子类。这些子类都必须重写以下方法:
`__len__`
:定义了数据集的规模;
`__getitem__`
:支持0到len(self)范围内的整数索引。
```
py
class
torch
.
utils
.
data
.
TensorDataset
(
*
tensors
)
...
...
@@ -31,7 +33,7 @@ class torch.utils.data.ConcatDataset(datasets)
class
torch
.
utils
.
data
.
Subset
(
dataset
,
indices
)
```
用索引指定的数据集子集。
用索引指定的数据集子集。
参数:
...
...
@@ -52,7 +54,7 @@ class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=
*
**num_workers**
(
[
_int_
](
https://docs.python.org/3/library/functions.html#int
"(in Python v3.7)"
)
_,_
_可选_) – 加载数据的子进程数量。0表示主进程加载数据(默认:
`0`
)。
*
**collate_fn**
(_可调用_ _,_ _可选_)– 归并样例列表来组成小批。
*
**pin_memory**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
_,_
_可选_)– 如果设置为
`True`
,数据加载器会在返回前将张量拷贝到CUDA锁页内存。
*
**drop_last**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
_,_
_可选_
)– 如果数据集的大小不能不能被批大小整除,该选项设为
`True`
后不会把最后的残缺批作为输入;如果设置为
`False`
,最后一个批将会稍微
小一点。(默认:
`False`
)
*
**drop_last**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
_,_
_可选_
)– 如果数据集的大小不能不能被批大小整除,该选项设为
`True`
后不会把最后的残缺批作为输入;如果设置为
`False`
,最后一个批将会稍微
小一点。(默认:
`False`
)
*
**timeout**
(_数值_ _,_ _可选_) – 如果是正数,即为收集一个批数据的时间限制。必须非负。(默认:
`0`
)
*
**worker_init_fn**
(_可调用_ _,_ _可选_)– 如果不是
`None`
,每个worker子进程都会使用worker id(在
`[0, num_workers - 1]`
内的整数)进行调用作为输入,这一过程发生在设置种子之后、加载数据之前。(默认:
`None`
)
...
...
@@ -98,7 +100,7 @@ class torch.utils.data.SequentialSampler(data_source)
class
torch
.
utils
.
data
.
RandomSampler
(
data_source
,
replacement
=
False
,
num_samples
=
None
)
```
随机采样
元素。如果replacement不设置,则从打乱之后的数据集采样。如果replacement设置了,那么用户可以指定
`num_samples`
来采样。
随机采样元素。如果replacement不设置,则从打乱之后的数据集采样。如果replacement设置了,那么用户可以指定
`num_samples`
来采样。
参数:
...
...
@@ -123,7 +125,7 @@ class torch.utils.data.WeightedRandomSampler(weights, num_samples, replacement=T
参数:
*
**weights**
(_序列_) – 权重序列,不需要
和为1。
*
**weights**
(_序列_) – 权重序列,不需要和为1。
*
**num_samples**
(
[
_int_
](
https://docs.python.org/3/library/functions.html#int
"(in Python v3.7)"
)
) – 采样数。
*
**replacement**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
) – 如果是
`True`
,替换采样。否则不替换,即:如果某个样本索引已经采过了,那么不会继续被采。
...
...
@@ -137,7 +139,7 @@ class torch.utils.data.BatchSampler(sampler, batch_size, drop_last)
*
**sampler**
(
[
_Sampler_
](
#torch.utils.data.Sampler
"torch.utils.data.Sampler"
)
)– 基采样器。
*
**batch_size**
(
[
_int_
](
https://docs.python.org/3/library/functions.html#int
"(in Python v3.7)"
)
)– 小批的规模。
*
**drop_last**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
)– 如果设置为
`True`
,采样器会丢弃最后一个
不够
`batch_size`
的小批(如果存在的话)。
*
**drop_last**
(
[
_bool_
](
https://docs.python.org/3/library/functions.html#bool
"(in Python v3.7)"
)
)– 如果设置为
`True`
,采样器会丢弃最后一个不够
`batch_size`
的小批(如果存在的话)。
示例
...
...
@@ -164,4 +166,4 @@ class torch.utils.data.distributed.DistributedSampler(dataset, num_replicas=None
*
**dataset**
– 采样的数据集。
*
**num_replicas**
(_可选_)– 参与分布式训练的进程数。
*
**rank**
(_可选_)– num_replicas中当前进程的等级。
\ No newline at end of file
*
**rank**
(_可选_)– num_replicas中当前进程的等级。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录