Python Generator Output Flow From Tensor