def begin(self):
    self._iter_count = 0
    self._current_tensors = {tag: _as_graph_element(tensor) for (tag, tensor) in self._tensors.items()}
def begin(self):
    self._iter_count = 0
    self._current_tensors = {tag: _as_graph_element(tensor) for (tag, tensor) in self._tensors.items()}