transformer_lens.conversion_utils.conversion_steps.base_tensor_conversion module

class transformer_lens.conversion_utils.conversion_steps.base_tensor_conversion.BaseTensorConversion(input_filter: Callable | None = None, output_filter: Callable | None = None)

Bases: object

Base class for tensor conversions.

convert(input_value, *full_context)

handle_conversion(input_value, *full_context)

revert(input_value, *full_context)

Revert the conversion. For now, just return the input unchanged.