如何获得Tensor的类型

Question

我正在寻找类似于x.get_shape()的效果的东西,它将给出x的类型.如果没有功能我该怎么办呢？

Answer 1

您可以使用get_shape()来获取tensorflow变量的形状.

>>> x = tf.Variable(tf.random_normal([256, 100]))
>>> x.get_shape()
(256, 100)

您可以使用dtype属性来获取tensorflow变量的类型.

>>> x = tf.Variable(tf.random_normal([256, 100]))
>>> x.dtype
<dtype: 'float32_ref'>

您可以使用dtype的as_numpy_dtype属性将tf.dtype转换为numpy dtype.

>>> x = tf.Variable(tf.random_normal([256, 100]))
>>> x.dtype.as_numpy_dtype
<class 'numpy.float32'>

Answer 2

获得你可以做的类型

x.dtype