def ParsePairFromString(string): """Converts serialized DatumPairProto string to NumPy arrays. Args: string: Serialized DatumProto string. Returns: Two NumPy arrays. """ datum_pair = datum_pb2.DatumPairProto() datum_pair.ParseFromString(string) return DatumPairToArrays(datum_pair)
def ArraysToDatumPair(arr_1, arr_2): """Converts numpy arrays to DatumPairProto. Supports same formats as `ArrayToDatum`, see documentation therein. Args: arr_1: NumPy array of arbitrary shape. arr_2: NumPy array of arbitrary shape. Returns: datum_pair: DatumPairProto object. """ datum_pair = datum_pb2.DatumPairProto() datum_pair.first.CopyFrom(ArrayToDatum(arr_1)) datum_pair.second.CopyFrom(ArrayToDatum(arr_2)) return datum_pair