如何在张量流中使用张量的动态形状

Question

如何为以下计算构建张量流图？我现在的问题是如何使用张量A的形状信息，它具有可变的形状大小。

A = tf.placeholder(tf.float32, [None,10])
B = tf.Variable(tf.random_normal([10,20]))
C = tf.matmul(A, B)
D = tf.matmul(tf.transpose(C), A) # the value of A.shape[0]