a_array = np.array([[1, 2, 3], [4, 5, 6]])
b_list = [[1, 2, 3], [3, 4, 5]]
c_tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
d=c_tensor.get_shape()
e=d.as_list()
print("a:",type(a_array))
print("b_list:",type(b_list))
print("c_tensor:",type(c_tensor))
print("d",type(d))
print("e",type(e))
输出:
a: <class 'numpy.ndarray'>
b_list: <class 'list'>
c_tensor: <class 'tensorflow.python.framework.ops.Tensor'>
d <class 'tensorflow.python.framework.tensor_shape.TensorShape'>
e <class 'list'>
可以看出来x必须是一个张量才能使用get_shape()方法,再通过as_list()方法可将结果转化为列表。
最后结果是x的维度。