为什么标量的初始化用的是一个空list？这里一定要用int64吗？

Question

global_step = tf.Variable(tf.zeros([], tf.int64), name='global_step', trainable=False)

请问，在定义模型时给出的记录训练次数的变量，为什么作为一个标量的初始化用的是一个空list？
2. 这里一定要用int64吗？不可以是int32吗？

Answer 1

正十七 2019-10-18 22:24:17

1. 空list表明这个Variable的维度是0, 所以是标量。

2. int64的范围比int32大，一般情况下，训练步数不过亿的话，用int32没有问题。