transformer_lens.conversion_utils.conversion_steps.rearrange_tensor_conversion module¶

class transformer_lens.conversion_utils.conversion_steps.rearrange_tensor_conversion.RearrangeTensorConversion(pattern: str, input_filter: Callable | None = None, output_filter: Callable | None = None, **axes_lengths)¶

Bases: BaseTensorConversion

handle_conversion(input_value: Tensor, *full_context) → Tensor¶

revert(input_value: Tensor, *full_context) → Tensor¶: Revert the conversion. For now, just return the input unchanged.