tf计算矩阵维度