张量流中的线性模型



i试图在张曲流中生成一个简单的线性模型。这是代码...

N        = 400
features = 100
nSteps   = 1000
data = (np.random.randn(N, features), np.random.randint(0, 2, N))
W = tf.placeholder(tf.float32, shape=(features,1), name='W')
b = tf.placeholder(tf.float32, shape=(features,1), name='b')
d = tf.constant(data[0], dtype=tf.float32)
result = tf.add( tf.matmul(d, W), b)

事实证明,b的尺寸可能存在一些问题,但是据我所知,它们都可以...

不确定为什么这会引起错误。有人可以帮忙吗?

注意:

result = tf.matmul(d, W)

这还可以。

我已经检查了结果的形状,并且与b的形状相同。不确定问题是什么。

在线性模型中(即输出层中的一个单元),b应该是标量。

在数学上,对于单个观察,您有:result = WX + b,其中尺寸W [1 x特征],X [功能X 1]。然后,WX是标量。因此b应该是标量。

因此,您应该将b更改为以下内容,以获取正确的线性模型并使尺寸奏效:

b = tf.placeholder(tf.float32, shape=(1,1), name='b')

相关内容

  • 没有找到相关文章

最新更新