code: def test_zero_diag():
      ones = tf.ones((10, 10))
      ones_zd = zero_diag(ones)
      assert ones_zd.shape == (10, 10)
      assert float(tf.linalg.trace(ones_zd)) == 0
      assert float(tf.reduce_sum(ones_zd)) == 90
  
file_path: alibi-detect/alibi_detect/utils/tensorflow/tests/test_misc_tf.py
imports: from itertools import product
  import pytest
  import tensorflow
  from tensorflow.keras.layers import Dense, Input, InputLayer
  import numpy
  from alibi_detect.utils.tensorflow import zero_diag, quantile, subset_matrix
  from alibi_detect.utils.tensorflow.misc import clone_model
label_desc: Use image node_id 1 for calling a global function with example usage: test_zero_diag() without return types