tabensemb.model.TorchModelWrapper.wrap_forward#

method

TorchModelWrapper.wrap_forward()[source]#

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