d3rlpy.algos.IdentityTransformerActionSampler¶

class d3rlpy.algos.IdentityTransformerActionSampler(*args, **kwargs)[source]¶

Identity action-sampler.

This class implements identity function to process Transformer output. Sampled action is the exactly same as transformer_output.

Methods

__call__(transformer_output)[source]¶

Returns sampled action from Transformer output.

Parameters:: transformer_output (ndarray[tuple[int, ...], dtype[Any]]) – Output of Transformer algorithms.
Returns:: Sampled action.
Return type:: Union[ndarray[tuple[int, …], dtype[Any]], int]