tltorch.tensor_hooks.TensorDropout¶

class tltorch.tensor_hooks.TensorDropout(proba, min_dim=1, min_values=1, drop_test=False)[source]¶

Decomposition Hook for Tensor Dropout on FactorizedTensor

Parameters:

nameFactorizedTensor parameter on which to apply the dropout

probafloat, probability of dropout

min_dimint

Minimum dimension size for which to apply dropout. For instance, if a tensor if of shape (32, 32, 3, 3) and min_dim = 4 then dropout will not be applied to the last two modes.

Methods

__call__(module, input, factorized_tensor)

Call self as a function.

apply