Args: ds (DataFlow): input DataFlow. ``dp[idx]`` has to be an :class:`np.ndarray`. batch_size (int): batch size idx (int): ``dp[idx].shape`` will be used to group datapoints. Other components are assumed to be batch-able.
(self, ds, batch_size, idx)
| 203 | never get generated. |
| 204 | """ |
| 205 | def __init__(self, ds, batch_size, idx): |
| 206 | """ |
| 207 | Args: |
| 208 | ds (DataFlow): input DataFlow. ``dp[idx]`` has to be an :class:`np.ndarray`. |
| 209 | batch_size (int): batch size |
| 210 | idx (int): ``dp[idx].shape`` will be used to group datapoints. |
| 211 | Other components are assumed to be batch-able. |
| 212 | """ |
| 213 | super(BatchDataByShape, self).__init__(ds, batch_size, remainder=False) |
| 214 | self.idx = idx |
| 215 | |
| 216 | def reset_state(self): |
| 217 | super(BatchDataByShape, self).reset_state() |