如何在张量流中获取不同操作的总参数数？

Question

我知道如何在tensorflow中获得可训练变量数。但是，在某些情况下（seq2seq，视频字幕），train_op的工作方式与predict_op略有不同。我想知道有多少个参数与train_op相关，有多少个参数与Forecast_op相关？

这里是获取可训练变量参数数量的代码。

def get_num_params():
print('==='* 30)
num_params = 0
for variable in tf.trainable_variables():
    shape = variable.get_shape()
    print(variable.name, shape)
    num_params += reduce(mul, [dim.value for dim in shape], 1)
print('total param number', num_params)
print('===' * 30)