code: def multivariate_normal_kl_divergence(
      mu_0: Array,
      sigma_0: Numeric,
      mu_1: Array,
      sigma_1: Numeric,
  ) -> Array:
      """Compute the KL between 2 gaussian distrs with diagonal covariance matrices.
  
      Args:
        mu_0: array like of mean values for policy 0
        sigma_0: array like of std values for policy 0
        mu_1: array like of mean values for policy 1
        sigma_1: array like of std values for policy 1
  
      Returns:
        the kl divergence between the distributions.
      """
      warnings.warn(
          "Rlax multivariate_normal_kl_divergence will be deprecated."
          "Please use distrax.MultivariateNormalDiag.kl_divergence instead.",
          PendingDeprecationWarning,
          stacklevel=2,
      )
      return distrax.MultivariateNormalDiag(
          mu_0, sigma_0
      ).kl_divergence(distrax.MultivariateNormalDiag(mu_1, sigma_1))
  
file_path: rlax/rlax/_src/distributions.py
imports: import collections
  import warnings
  import chex
  import distrax
  import jax.numpy
label_desc: Use image node_id 11 for calling a global function with example usage: multivariate_normal_kl_divergence(mu_0, sigma_0, mu_1, sigma_1) and returns: distrax