def _recurrent_kernel_int2ext(
        self, recurrent_kernel_int: torch.tensor
    ) -> torch.Tensor:
        """
        >>> (); mod = sLSTMCell_cuda(
        ...     sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS
        (...)
        >>> torch.allclose(mod._recurrent_kernel_ext2int(mod._recurrent_kernel_int2ext(mod._recurrent_kernel)), mod._recurrent_kernel)
        True
        """

        return recurrent_kernel_int.reshape(
            self.config.num_heads,
            self.config.head_dim,
            self.config.num_gates,
            self.config.head_dim,
        )

        >>> (); mod = sLSTMCell_cuda(
        ...     sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS
        (...)
        >>> torch.allclose(mod._recurrent_kernel_ext2int(mod._recurrent_kernel_int2ext(mod._recurrent_kernel)), mod._recurrent_kernel)
        True
        
xlstm/blocks/slstm/cell.py