Is this similar to subset selection with the concrete distribution?