tensorflow | 维度转换

tf.shape(input,name = None)

a = tf.constant([i for i in range(20)],shape =[2,2,5])
with tf.Session() as sess:
    print (sess.run(tf.shape(a)))

tf.size(input,name = None)

a = tf.constant([i for i in range(20)],shape =[2,2,5])
with tf.Session() as sess:
    print (sess.run(tf.size(a)))

tf.rank(input, name=None)

a = tf.constant([i for i in range(20)],shape =[2,2,5])
with tf.Session() as sess:
    print (sess.run(tf.rank(a)))

tf.reshape(tensor, shape, name=None)

a = tf.constant([i for i in range(20)],shape =[2,2,5])
with tf.Session() as sess:
    print (sess.run(tf.reshape(a,shape = [5,2,2])))

tf.squeeze(input, squeeze_dims=None, name=None)

tf.expand_dims(input, dim, name=None)