def label_func(s, label=label):
                self.set_label(s, label)
def deeplift_tensor_grad(grad):
    return_grad = complex_module_gradients[-1]
    del complex_module_gradients[-1]
    return return_grad