Coverage for transformer_lens/conversion_utils/conversion_steps/zeros_like_conversion.py: 100%

1import torch 

2 

3from .base_tensor_conversion import BaseTensorConversion 

4 

5 

6class ZerosLikeConversion(BaseTensorConversion): 

7 def handle_conversion(self, input_value: torch.Tensor, *full_context) -> torch.Tensor: 

8 return torch.zeros_like(input_value) 

9 

10 def __repr__(self): 

11 return "Is a zeros_like operation"