tf.newaxis的主要用途是增加一个维度,位置不同,增加的维度是第几维也相对不一样,可以解决用训练好的神经网络模型进行预测时的维度不匹配问题。
img_arr = np.array(img.convert('L'))
print("img_arr:", img_arr.shape)
x_predict = img_arr[tf.newaxis, ..
tf.newaxis的主要用途是增加一个维度,位置不同,增加的维度是第几维也相对不一样,可以解决用训练好的神经网络模型进行预测时的维度不匹配问题。
img_arr = np.array(img.convert('L'))
print("img_arr:", img_arr.shape)
x_predict = img_arr[tf.newaxis, ..
1073
7596
1872

被折叠的 条评论
为什么被折叠?