tabensemb.model.PytorchTabularWrapper.wrap_forward#

method

PytorchTabularWrapper.wrap_forward()[source]#

Override the forward method of a torch.nn.Module to record hidden representations.