当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.io.ResizeIter用法及代码示例


用法:

class mxnet.io.ResizeIter(data_iter, size, reset_internal=True)

参数

  • data_iter(DataIter) - 要调整大小的数据迭代器。
  • size(int) - 每个时期要调整大小的批次数。
  • reset_internal(bool) - 是否在ResizeIter.reset 上重置内部迭代器

基础:mxnet.io.io.DataIter

将数据迭代器的大小调整为给定的批次数。

例子

>>> nd_iter = mx.io.NDArrayIter(mx.nd.ones((100,10)), batch_size=25)
>>> resize_iter = mx.io.ResizeIter(nd_iter, 2)
>>> for batch in resize_iter:
...     print(batch.data)
[<NDArray 25x10 @cpu(0)>]
[<NDArray 25x10 @cpu(0)>]

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.io.ResizeIter。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。