當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。