将张量转换为列表

时间:2018-08-05 23:48:59

标签: python-3.x tensorflow tensor

我正在尝试从操作中获取张量的列表(因为tf.matmul或tf.concat返回张量的张量而不是张量的列表)我想使用它,因此我可以在模型内部链接操作而无需运行会话。 我发现tf.create_partitioned_variables很有用,但是我无法弄清楚如何使用它,因为尽管张量的形状是(2 ,?),尽管指定了要输入的ph的形状,甚至使用tf.reshape。

x_ph = tf.placeholder(dtype=tf.float32, shape=[2, 9])
model_output = model(x_ph)

def model(x):
     x = tf.concat([weight_ones, x], axis=1)
     var = tf.create_partitioned_variables([2,10], [2,1], x) # this wont work
     result = tf.matmul(var, some_other_var)
     #then i want to feed the result into another tf.matmul

有什么办法可以完成我想做的事情?

0 个答案:

没有答案