cr.sparse.take_along_rows

cr.sparse.take_along_rows(X, indices)[source]

Picks K entries from each row of X specified by indices matrix