mindspore.ops.get_vm_impl_fn

mindspore.ops.get_vm_impl_fn(prim)

通过Primitive对象或Primitive名称，获取虚拟实现函数。

参数：

• prim (Union[Primitive, str]) - 算子注册的Primitive对象或名称。

说明

该机制目前适用于调试。

返回：

函数，虚拟实现函数。

样例：

>>> from mindspore.ops import vm_impl_registry, get_vm_impl_fn
...
>>> @vm_impl_registry.register("Type")
... def vm_impl_dtype(self):
...   def vm_impl(x):
...     return type(x)
...   return vm_impl
...
>>> fn = get_vm_impl_fn("Type")
>>> out = fn(1.0)
>>> print(out)
<class 'float'>