Deep clustering with concrete k-means