tensorflow之reshape

reshape可以对shape进行重新分配

示例:

a_array=tf.constant([[[1,2,3], [4,5,6]], [[1,1,1], [1,1,1]]])
print(a_array)
print(tf.reshape(a_array,[2,6]))

 输出:

猜你喜欢

转载自blog.csdn.net/g0415shenw/article/details/86242266