maldil/JavaFyPy

yield expression and yield from expression

Opened this issue · 0 comments

    class Pop(nn.Module):
        def forward(self, input):
            foo = yield pop("foo")
            return foo # noqa
    def __iter__(self) -> Iterator[Tensor]:
        if self.atomic:
            yield self.tensor
        else:
            yield from self.tensors