    def get_input_dim(self) -> int:
        """
        Returns the expected final dimension of the `sequence_tensor`.
        """
        raise NotImplementedError
Returns the expected final dimension of the `sequence_tensor`.
allennlp/modules/span_extractors/span_extractor.py