Fast randomly sample a varied number of ones by row from one hot encoded matrix

Question

I have a one hot encoded M x N matrix, A, with the following properties:

• 1 or more columns in each row can eq 1
• Every column in t