tabensemb.model.AbstractWrapper.wrap_forward#

method

AbstractWrapper.wrap_forward()[source]#

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