FunctionRuntime¶
- class madspace.FunctionRuntime(*args, **kwargs)¶
Bases:
pybind11_objectOverloaded function.
__init__(self: madspace._madspace_py.FunctionRuntime, function: madspace._madspace_py.Function) -> None
__init__(self: madspace._madspace_py.FunctionRuntime, function: madspace._madspace_py.Function, context: madspace._madspace_py.Context) -> None
- call(self: madspace._madspace_py.FunctionRuntime, arg0: collections.abc.Sequence[object]) list[madspace._madspace_py.Tensor]¶
- call_backward(self: madspace._madspace_py.FunctionRuntime, arg0: collections.abc.Sequence[object], arg1: collections.abc.Sequence[object], arg2: collections.abc.Sequence[bool]) tuple[list[madspace._madspace_py.Tensor | None], list[tuple[str, madspace._madspace_py.Tensor | None]]]¶
- call_with_grad(self: madspace._madspace_py.FunctionRuntime, arg0: collections.abc.Sequence[object], arg1: collections.abc.Sequence[bool]) tuple[list[madspace._madspace_py.Tensor], list[madspace._madspace_py.Tensor | None], list[bool]]¶
- release_inputs(self: madspace._madspace_py.FunctionRuntime) None¶