【发布时间】:2017-10-12 14:40:10
【问题描述】:
我有一个占位符,其形状取决于另一个占位符。 在占位符初始化期间如何连接它们?
nUsers = tf.placeholder(tf.float32, [None, 1])
p = tf.placeholder(tf.float32, [None, 10 , ???] )
...在 ???我需要将 nUsers 中存在的数字给定的大小用于该批次的该项目。
【问题讨论】:
我有一个占位符,其形状取决于另一个占位符。 在占位符初始化期间如何连接它们?
nUsers = tf.placeholder(tf.float32, [None, 1])
p = tf.placeholder(tf.float32, [None, 10 , ???] )
...在 ???我需要将 nUsers 中存在的数字给定的大小用于该批次的该项目。
【问题讨论】:
您可以使用x.get_shape()(check the difference 在tf.shape(x) 之间)轻松获得张量的形状。所以你需要这样的东西:
import tensorflow as tf
import numpy as np
nUsers = tf.placeholder(tf.float32, [None, 1])
p = tf.placeholder(tf.float32, [None, 10 , nUsers.get_shape()[0]])
with tf.Session() as sess:
print sess.run(p, {nUsers: np.ones((2, 1)), p: np.ones((3, 10, 2))})
【讨论】: