我是Tensorflow的新手。我试图将张量列表放入tf.train.batch()
函数,但我得到了ValueError (too many values to unpack (expected 2))
。这是我的代码,错误来自第二行。
g = tf.unstack(data, num = 60366, axis = 0)
X_mb, _ = tf.train.batch(g, 32, capacity = 60366)
我的数据是[60366,39]矩阵。
有人能帮助我吗? THX!
答案 0 :(得分:2)
看起来你需要这个:
X_mb, *_ = tf.train.batch(g, 32, capacity = 60366)
或者这个:
X_mb = tf.train.batch(g, 32, capacity = 60366)[0]