def forward(
self,
input: Tensor) -> Generator[stash, None, Tensor]: # type: ignore
yield stash('skip', input)
return input
def forward(self, tensor: Tensor) -> Tensor: # type: ignore
yield stash('identity', tensor)
return tensor
def forward(self, x):
yield stash('hello', x)
return self.conv(x)
def forward(self, input):
yield stash('foo', input)
return input * 2
def forward(self, input):
foo = yield stash('foo')
return input + foo
def forward(self, input):
yield stash('bar', input)
return input
def forward(self, input):
yield stash('skip', input)
return input
def forward(self, input):
yield stash('none', None)
return input
def forward(self, input):
yield stash('1to3', input)
output = self.conv(input)
return output
def forward(self, tensor):
yield stash('identity', tensor)
return tensor
def forward(self, input: torch.Tensor):
yield stash("skip", input)
return input # noqa using yield together with return