transformer_lens.conversion_utils.conversion_steps.repeat_tensor_conversion module

class transformer_lens.conversion_utils.conversion_steps.repeat_tensor_conversion.RepeatTensorConversion(pattern: str, input_filter: Callable | None = None, output_filter: Callable | None = None, **axes_lengths)

Bases: BaseTensorConversion

handle_conversion(input_value: Tensor, *full_context) → Tensor