def yaml(self):
        return yaml.dump(self.serialized, default_flow_style=False,
                         encoding=None)
def deeplift_tensor_grad(grad):
    return_grad = complex_module_gradients[-1]
    del complex_module_gradients[-1]
    return return_grad