pymoose.output

pymoose.output(tag, value, placement=None)

Tag an output of a computation.

This op is similar to identity, but additionally tags its value. It can be used to pin an output to a particular placement without writing it to placement storage. It’s also useful for maintaining the ordering of computation outputs in PyMoose, since Moose compilation generally doesn’t preserve output order.

Parameters

• tag – A tag to associate with the output, useful for reconstructing the original order of outputs.

• value – The Moose value to output.

• placement – An optional Placement to pin this operation to.

Returns

The tagged value.