How to use the tfrecord.writer.TFRecordWriter function in tfrecord

def write(self, datum: typing.Dict[str, typing.Tuple[typing.Any, str]]) -> None:
        """Write an example into tfrecord file.

        Params:
        -------
        datum: dict
            Dictionary of tuples of form (value, dtype). dtype can be
            "byte", "float" or "int".
        """
        record = TFRecordWriter.serialize_tf_example(datum)
        length = len(record)
        length_bytes = struct.pack("