Python 使用TensorFlow 2记录训练期间的体重标准

我正在使用tensorflow训练一个神经网络，我想监控一些层的L2范数。

---

例如，我在tensorflow 1中找到了这样做的方法，但在tensorflow 2中找不到同样的方法。我不知道我该怎么做。谢谢

Tensorflow 2提供了Tensorflow 1的兼容版本。 请查找以下用于计算非零权重的代码

import tensorflow as tf
import numpy as np

#Replace tf.trainable_variables() with tf.compat.v1.trainable_variables
tvars = tf.compat.v1.trainable_variables()
nonzero_parameters = np.sum([np.count_nonzero(var) for var in tvars])