yourTensor.get_shape().as_list()

黄世宇/Shiyu Huang's Personal Page:​​https://huangshiyu13.github.io/​