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[Any, dtype[Any]]) – Output of Transformer algorithms.

Returns:

Sampled action.

Return type:

Union[ndarray[Any, dtype[Any]], int]