如何在“ Tensorflow”中将一个numpy数组转换为张量?

时间:2018-11-09 09:29:15

标签: numpy tensorflow

使用张量流完成训练后,我尝试测试一些学习网络。

但是我的测试图像是通道1的[512 512 1]数据(水平512和垂直512)。

我将图像数据更改为一个numpy数组。

张量网络应该是[? 512 512 1]看起来像这样。

如何将numpy数组转换为张量? ([512 512 1]-> [?512 512 1])

1 个答案:

答案 0 :(得分:2)

您只需附加一个维度

arr = your_image # [512, 512, 1]
new_arr = np.expand_dims(arr, 0)

tensor = tf.convert_to_tensor(new_arr)

现在您可以使用feed dict或其他东西。