mindspore.ops.get_vm_impl_fn
- mindspore.ops.get_vm_impl_fn(prim)[源代码]
通过Primitive对象或Primitive名称,获取虚拟实现函数。
参数:
prim (Union[Primitive, str]) - 算子注册的Primitive对象或名称。
Note
该机制目前适用于调试。
返回:
函数,虚拟实现函数。
样例:
>>> from mindspore.ops import vm_impl_registry >>> from mindspore.ops.vm_impl_registry import 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'>