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

来源:7-13 计算图输入定义

慕虎9426780

2019-10-03

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

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

写回答

1回答

正十七

2019-10-18

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

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

0
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程