代码之家  ›  专栏  ›  技术社区  ›  m33n

张量流性能:Numpy矩阵o TF矩阵?

  •  0
  • m33n  · 技术社区  · 7 年前

    我有以下代码:

    with tf.Session() as sess:
        sess.run(init_vars)
        cols = sess.run(tf.shape(descriptors)[1])
        descriptor_matrix = np.zeros((n_batches*batch_size, cols))
        while True:
            batch_descriptor = sess.run(descriptors, feed_dict={dropout_prob: 1})
            descriptor_matrix[i:i+elements_in_batch] = np.array(batch_descriptor)
    

    我混合了张量和numpy向量。这对性能有重要影响吗?为什么会这样?我应该用张量来代替吗?

    1 回复  |  直到 7 年前
        1
  •  2
  •   H.H    7 年前

    混合张量和numpy是没有问题的。 但是在“graph”(sess.run())之外执行处理是不好的。

    Tensorflow/XLA编译器,用于生成“图形”以优化性能。

    但是,处理外部图形(sess.run())是普通的python代码。