code: def chol(sqrt: tf.Tensor) -> tf.Tensor:
      return tf.stack([tf.linalg.diag(sqrt[:, i]) for i in range(Ln)])
  
imports: import numpy
  import pytest
  import tensorflow
  from numpy.testing import assert_allclose
  import gpflow
  from gpflow import Parameter
  from gpflow.base import AnyNDArray, MeanAndVariance
  from gpflow.conditionals import conditional
  from gpflow.config import default_float
  from gpflow.kernels import Kernel
  from gpflow.utilities.bijectors import triangular
label_desc: Use image node_id 6 for calling a global function with example usage: chol(sqrt) and returns: tf