Python tf.experimental.unregister_dispatch_for用法及代码示例

注销使用 @dispatch_for_* 注册的函数。

用法

tf.experimental.unregister_dispatch_for(
    dispatch_target
)

这主要用于测试目的。

例子：

# Define a type and register a dispatcher to override `tf.abs`:
class MyTensor(tf.experimental.ExtensionType):
  value:tf.Tensor
@dispatch_for_api(tf.abs)
def my_abs(x:MyTensor):
  return MyTensor(tf.abs(x.value))
tf.abs(MyTensor(5))
MyTensor(value=<tf.Tensor:shape=(), dtype=int32, numpy=5>)

# Unregister the dispatcher, so `tf.abs` no longer calls `my_abs`.
unregister_dispatch_for(my_abs)
tf.abs(MyTensor(5))
Traceback (most recent call last):

ValueError:Attempt to convert a value ... to a Tensor.