分类：形状

2020
10-08

TensorFlow中如何确定张量的形状实例

我们可以使用tf.shape()获取某张量的形状张量。importtensorflowastfx=tf.reshape(tf.range(1000),[10,10,10])sess=tf.Session()sess.run(tf.shape(x))Out[1]:array([10,10,10])我们可以使用tf.shape()在计算图中确定改变张量的形状。high=tf.shape(x)[0]//2width=tf.shape(x)[1]*2x_reshape=tf.reshape(x,[high,width,-1])sess.run(tf.shape(x_reshape))Out:array([5,20,10])我们可以使用tf.shap... 继续阅读 >