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