def sample(
     self,
     qubits: Sequence['cirq.Qid'],
     repetitions: int = 1,
     seed: 'cirq.RANDOM_STATE_OR_SEED_LIKE' = None,
 ) -> np.ndarray:
     indices = [self.qubit_map[q] for q in qubits]
     return sim.sample_density_matrix(
         self.target_tensor,
         indices,
         qid_shape=tuple(q.dimension for q in self.qubits),
         repetitions=repetitions,
         seed=seed,
     )
    def sample(
        self,
        axes: Sequence[int],
        repetitions: int = 1,
        seed: 'cirq.RANDOM_STATE_OR_SEED_LIKE' = None,
    ) -> np.ndarray:
        """Samples the density matrix.

        Args:
            axes: The axes to sample.
            repetitions: The number of samples to make.
            seed: The random number seed to use.
        Returns:
            The samples in order.
        """
        return sim.sample_density_matrix(
            self._density_matrix,
            axes,
            qid_shape=self._qid_shape,
            repetitions=repetitions,
            seed=seed,
        )