transformer_lens.conversion_utils.conversion_steps.callable_tensor_conversion module¶
- class transformer_lens.conversion_utils.conversion_steps.callable_tensor_conversion.CallableTensorConversion(convert_callable: Callable)¶
Bases:
BaseTensorConversion- handle_conversion(input_value: dict, *full_context)¶