gpt4 book ai didi

python - TensorFlow 初始化 Tensor of ones

转载 作者:太空狗 更新时间:2023-10-29 21:45:53 24 4
gpt4 key购买 nike

假设我有一个张量

X = tf.placeholder("float", [None, 5])

所以我知道列数但不知道行数。我需要初始化一个维度为 nrows x 1

的向量

现在下面的代码块不起作用,

o = tf.ones(shape=(tf.shape(X)[0], 1))
==> TypeError: List of Tensors when single Tensor expected

也没有,

o = tf.ones(shape=(X.get_shape()[0].value, 1))
==> TypeError: Input 'dims' of 'Fill' Op has type
string that does not match expected type of int32.

现在,我发现解决这个问题的一种方法是让我的向量成为一个占位符,

o = tf.placeholder(dtype=tf.float32, shape=[None, 1])

并在我的 feed_dict 中传入一个适当大小的 numpy 数组。但是这个解决方案让我觉得不够优雅,而不是占位符的预期用途。我在这里可能是错的,但肯定有更好的方法。

最佳答案

解决你的问题的方法是使用 tf.pack 操作:

o = tf.ones(shape=tf.pack([tf.shape(X)[0], 1]))

出现错误的原因是 TensorFlow shape 应该是整数列表或张量 link . tf.pack 可以轻松地将整数列表和/或 TensorFlow 标量转换为 Tensor 对象。

关于python - TensorFlow 初始化 Tensor of ones,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33711427/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com