在给shap传递参数的时候[1] 和 [1, ]是不是等价

来源:4-15 综合小练习:梯度下降解决线性回归(中)

Giroy3925940

2018-04-10

写回答

1回答

Oscar

2018-04-10

我想你说的是这句中 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))

 也是等价的。

具体可以看看我做的小测试:

//img.mukewang.com/szimg/5acc91860001444907120524.jpg

1
1
Giroy3925940
非常感谢!
2018-04-10
共1条回复

基于Python玩转人工智能最火框架 TensorFlow应用实践

机器学习入门,打牢TensorFlow框架应用是关键!

2214 学习 · 688 问题

查看课程