为了账号安全,请及时绑定邮箱和手机立即绑定

Tensorflow 性能:Numpy 矩阵或 TF 矩阵?

Tensorflow 性能:Numpy 矩阵或 TF 矩阵?

LEATH 2021-07-13 13:08:51
我有以下代码: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 回答

  • 1 回答
  • 0 关注
  • 180 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信