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