tensorplay.library.register_autograd

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