Searched refs:unbatch_level (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/utils/data/datapipes/iter/ |
| D | grouping.py | 133 def __init__(self, datapipe: IterDataPipe, unbatch_level: int = 1):
135 self.unbatch_level = unbatch_level
139 yield from self._dive(element, unbatch_level=self.unbatch_level)
141 def _dive(self, element, unbatch_level): argument
142 if unbatch_level < -1:
144 if unbatch_level == -1:
147 yield from self._dive(item, unbatch_level=-1)
150 elif unbatch_level == 0:
155 yield from self._dive(item, unbatch_level=unbatch_level - 1)
|
| D | combinatorics.py | 107 unbatch_level: int = 0,
114 if unbatch_level == 0:
117 self.datapipe = datapipe.unbatch(unbatch_level=unbatch_level)
|
| /external/pytorch/torch/utils/data/ |
| D | standard_pipes.ipynb | 67 …" - `unbatch_level:int = 0` if specified calls `unbatch(unbatch_level=unbatch_level)` on source d…
150 …ssible to unbatch source data before applying the new batching rule using `unbatch_level` argument"
169 "dp = ExampleIterPipe(30).batch(3).batch(2).batch(10, unbatch_level=-1)\n",
185 " `unbatch_level:int = 1`\n",
213 …"By default unbatching is applied only on the first layer, to unbatch deeper use `unbatch_level` a…
232 "dp = ExampleIterPipe(40).batch(2).batch(4).batch(3).unbatch(unbatch_level = 2)\n",
239 "Setting `unbatch_level` to `-1` will unbatch to the lowest level"
258 "dp = ExampleIterPipe(40).batch(2).batch(4).batch(3).unbatch(unbatch_level = -1)\n",
528 …" - `unbatch_level:int = 0` if specified calls `unbatch(unbatch_level=unbatch_level)` on source d…
583 "To shuffle elements across batches use `shuffle(unbatch_level)` followed by `batch` pattern "
[all …]
|
| D | dataframes_pipes.ipynb | 438 "dp = dp.batch(5).batch(3).batch(1).unbatch(unbatch_level = 3)\n",
|
| /external/pytorch/test/ |
| D | test_datapipe.py | 1861 unbatch_dp = input_dp.unbatch(unbatch_level=2)
1868 unbatch_dp = input_dp.unbatch(unbatch_level=-1)
1876 unbatch_dp = input_dp.unbatch(unbatch_level=-2)
1882 unbatch_dp = input_dp.unbatch(unbatch_level=5)
1888 unbatch_dp = input_dp.unbatch(unbatch_level=-1)
|