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'>