在给shap传递参数的时候[1] 和 [1, ]是不是等价
来源:4-15 综合小练习:梯度下降解决线性回归(中)
Giroy3925940
2018-04-10
写回答
1回答
-
我想你说的是这句中 tf.random_uniform 方法的参数 shape 吧?
W = tf.Variable(tf.random_uniform([1], -1.0, 1.0))
TensorFlow 的文档里写了:shape: A 1-D integer Tensor or Python array.
所以这个 shape 是一个 1 维的矢量(vector)。
如果你写成:
W = tf.Variable(tf.random_uniform([1,], -1.0, 1.0))
也是等价的。
具体可以看看我做的小测试:
112018-04-10
相似问题