Python annotate示例

示例#1

 def forward(self, x):
     with annotate(aaa='a', bbb=['b', 'c']):
         h = self.conv(x)
     h = self.conv2(h)
     with annotate(zzz=99, yyy=[9, 9]):
         h = self.linear(h)
         h = self.linear2(h)
     return h

示例#2

 def forward(self, x):
     with annotate(aaa='a'):
         h = self.conv(x)
     h = self.linear(h)
     return h