def forward(self, x): y = annotate(x, TensorType((1, 2, 3, Dyn))) return torch.add(x, y)
def forward(self, x): y = annotate(x, int) return torch.add(x, y)