Module: oryx.core.state

View source on GitHub

Module for stateful functions.

Modules

api module: Module for single-dispatch functions for handling state.

function module: Module for transforming functions into FunctionModules.

module module: Contains highest-level abstractions for the stateful function API.

Classes

class ArraySpec: Encapsulates shape and dtype of an abstract array.

class Module: Encapsulates a parameterized function, along with updates to its state.

class Shape: Encapsulates shape and dtype of an abstract array.

Functions

assign(...): Assigns a value to a variable.

call(...)

call_and_update(...)

init(...): Transforms an object into a function that initializes a module.

make_array_spec(...)

spec(...): A general purpose transformation for getting the output shape and dtype of an object.

update(...)

variable(...): Tags a value as a variable.

{
 layer_cau: <function layer_cau_kwargs_rule>
}