import os
os.environ[‘TF_CPP_MIN_LOG_LEVEL‘] = ‘2‘
import tensorflow as tf
def tensor_shape_demo():
a_p = tf.placeholder(dtype=tf.float32,shape=[None,None])
b_p = tf.placeholder(dtype=tf.float32,shape=[None,5])
print(‘======================Static shape modification‘)
print(‘The shape before modification:‘)
print(‘a_p:‘,a_p)
print(‘b_p:‘,b_p)
print(‘The shape after modification:‘)
a_p.set_shape([3,4])
b_p.set_shape([2,5])
print(‘a_p:‘,a_p)
print(‘b_p:‘,b_p)
print(‘======================Dynamic shape modification‘)
c_p = tf.placeholder(dtype=tf.float32,shape=[3,4])
print(‘The shape before modification:‘)
print(‘c_p:‘,c_p)
print(‘The shape after modification:‘)
c_p = tf.reshape(c_p,shape=[2,6])
print(‘c_p:‘,c_p)
c_p = tf.reshape(c_p,shape=[4,3])
print(‘c_p:‘,c_p)
c_p = tf.reshape(c_p,shape=[3,4,1])
print(‘c_p:‘,c_p)
if __name__ == ‘__main__‘:
tensor_shape_demo()
原文:https://www.cnblogs.com/monsterhy123/p/13066715.html