oryx.core.sow

Marks a value with a name and a tag.

View aliases

Main aliases

oryx.core.interpreters.harvest.sow

oryx.core.sow(
    value, *, tag='strict', name=None, mode='strict', key=None
)

Args
value	A JAX value to be tagged and named. tag (str): a string representing the tag of the sown value. name (str): a string representing the name to sow the value with. mode (str): The mode by which to sow the value. There are three options: 1. strict - if another value is sown with the same name and tag in the same context, harvest will throw an error. 2. clobber - if another is value is sown with the same name and tag, it will replace this value 3. append - sown values of the same name and tag are appended to a growing list. Append mode assumes some ordering on the values being sown defined by data-dependence.
key	an optional JAX value that will be tied into the sown value.

Returns
The original value that was passed in.