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