specified_prob_mask_generator function

A torch::nn_module() Representing a specified_prob_mask_generator