def get_replacement(
        self,
        node: _torch.fx.Node,
        output_shape: _Tuple[int],
        modules: _Dict[str, _torch.nn.Module],
        env: _Dict[str, _Any],
    ) -> _Any:
        """
        Computes the transformed node as a result of the transformation given
        input node and expected output shape of this node.
        """
        pass

        Computes the transformed node as a result of the transformation given
        input node and expected output shape of this node.
        
coremltools/optimize/torch/_utils/transforms.py