Python tf.Tensor.ref用法及代碼示例

用法

ref()

返回此張量的可散列引用對象。

此 API 的主要用例是將張量放入集合/字典中。我們不能將張量放在集合/字典中，因為從 Tensorflow 2.0 開始，tensor.__hash__() 不再可用。

以下將引發從 2.0 開始的異常

x = tf.constant(5)
y = tf.constant(10)
z = tf.constant(10)
tensor_set = {x, y, z}
Traceback (most recent call last):

TypeError:Tensor is unhashable. Instead, use tensor.ref() as the key.
tensor_dict = {x:'five', y:'ten'}
Traceback (most recent call last):

TypeError:Tensor is unhashable. Instead, use tensor.ref() as the key.

相反，我們可以使用 tensor.ref() 。

tensor_set = {x.ref(), y.ref(), z.ref()}
x.ref() in tensor_set
True
tensor_dict = {x.ref():'five', y.ref():'ten', z.ref():'ten'}
tensor_dict[y.ref()]
'ten'

此外，參考對象提供了返回原始張量的.deref() 函數。

x = tf.constant(5)
x.ref().deref()
<tf.Tensor:shape=(), dtype=int32, numpy=5>