将张量变量(不是常数)转换为numpy数组?

时间:2018-07-23 04:19:14

标签: tensorflow deep-learning conv-neural-network tensor

我正在将嵌入从CNN模型的最后一个完全连接的层转换为numpy数组。当前,嵌入是可变张量,该张量不是恒定的。我想知道如何将其转换为numpy数组。

1 个答案:

答案 0 :(得分:0)

您应该在会话中将张量值分配给变量! 这样的事情。这是为了从一开始就消除瓶颈值。希望这会有所帮助。

with tf.Session(graph=graph) as sess:
    im_input = graph.get_tensor_by_name('inception/ExpandDims:0')
    Bottle_neck_values = []
    m = train.shape[0]
    indices = np.arange(m)
    for index in tqdm(range(0, m, 1)):
        batch = indices[index:index + 1]
        Bottle_neck_values.append(sess.run(bottleneck, feed_dict= 
                                                   {im_input:train[batch]}))
    df = pd.DataFrame(np.array(np.squeeze(Bottle_neck_values)))
    df.to_csv("./bottleneck/Bottle_neck_values.csv", index=False)