def get_output_from_indices(self, indices):
        """Get output from indices.

        Args:
            indices: Tensor of shape [B, T]

        Returns:
            Tensor of shape [B, D, T]
        """
        z_q = self.get_codes_from_indices(indices)
        return self.out_project(z_q.transpose(0, 2, 1)).transpose(0, 2, 1)
Get output from indices.

        Args:
            indices: Tensor of shape [B, T]

        Returns:
            Tensor of shape [B, D, T]
        
mlx_audio/tts/models/spark/modules/residual.py