mixmo.augmentations.mixing_blocks._n_mix¶

mixmo.augmentations.mixing_blocks._n_mix(method, lams, input_size, config_mix)[source]¶

Computes masks for M>2 inputs (requires lam tuples)

method: string, mixing to use lam: float

masks: list of torch.Tensor. The masks used for mixing.

lams: list of torch.Tensor. The lams ratio on the final masks.