neuralogic.nn.torch_function