How to write TFRecord in Pytorch?

1

I want to convert below some lines of TensorFlow to Pytorch which are related to TFRecord.

Here are the lines of codes:

tf.train.Feature used to create integer or byte feature)]

feature_integer = tf.train.Feature(int64_list=tf.train.Int64List(value=list(values)))

feature_byte = tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))

tf_example = tf.train.Example(features=tf.train.Features(feature=features))
tf.python_io.TFRecordWriter(tf_example.SerializeToString())