Exemple #1
0
 def forward(self, input):
     yield stash("none", None)
     return input
Exemple #2
0
 def forward(self, input):
     yield stash("1to3", input)
     output = self.conv(input)
     return output
Exemple #3
0
 def forward(self, input):
     yield stash("foo", input)
     return input * 2
Exemple #4
0
 def forward(self, input):
     yield stash("skip", input)
     return input
Exemple #5
0
 def forward(self, input):
     yield stash("bar", input)
     return input
Exemple #6
0
 def forward(self, input):
     foo = yield stash("foo")
     return input + foo
Exemple #7
0
 def forward(self, x):
     yield stash("hello", x)
     return self.conv(x)