tensorplay.library.register_vmap

tensorplay.library.register_vmap(op: str | CustomOpDef, func: Callable[[...], Any] | None = None, /, *, lib: Any = None) → Callable[[Callable[[...], Any]], Callable[[...], Any]][source]