yield expression and yield from expression
Opened this issue · 0 comments
maldil commented
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