Python tf.TensorShape.__eq__用法及代码示例

用法

__eq__(
    other
)

如果 self 等效于 other ，则返回 True。

它首先尝试将 other 转换为 TensorShape 。 TypeError 转换失败时抛出。否则，它会比较 TensorShape 维度中的每个元素。

• 两个完全已知的形状，如果每个元素相等，则返回 True。

>>> t_a = tf.TensorShape([1,2])
>>> a = [1, 2]
>>> t_b = tf.TensorShape([1,2])
>>> t_c = tf.TensorShape([1,2,3])
>>> t_a.__eq__(a)
True
>>> t_a.__eq__(t_b)
True
>>> t_a.__eq__(t_c)
False

• 两个 Partially-known 形状，返回 False。

>>> p_a = tf.TensorShape([1,None])
>>> p_b = tf.TensorShape([2,None])
>>> p_a.__eq__(p_b)
False
>>> t_a.__eq__(p_a)
False

• 两个未知形状，返回 True。

>>> unk_a = tf.TensorShape(None)
>>> unk_b = tf.TensorShape(None)
>>> unk_a.__eq__(unk_b)
True
>>> unk_a.__eq__(t_a)
False