acoustic-carpenter-78188
09/11/2023, 12:38 AMtf.tensor as a native Flyte type.
API reference for serialization and deserialization of tensors:
• Serialization: https://www.tensorflow.org/api_docs/python/tf/io/serialize_tensor
• Deserialization: https://www.tensorflow.org/api_docs/python/tf/io/parse_tensor
More on TypeTransfomers can be found here.
Related PR that adds PyTorch tensor and module as Flyte types: flyteorg/flytekit#1032
flyteorg/flyteacoustic-carpenter-78188
09/11/2023, 12:38 AM