oryx.core.state.spec

A general purpose transformation for getting the output shape and dtype of an object.

View aliases

Main aliases

oryx.core.state.api.spec

oryx.core.state.spec(
    obj
) -> oryx.core.ppl.LogProbFunction

spec is used to query the output shapes and dtypes of objects like Modules or functions.

Args
obj	The object whose spec will be queried.

Returns
A function when passed in values or specs will return an output spec.