dmlc / dmlc/dlpack

[Feature Request] protobuf dlpack support

Open
#112 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
C++
Stars
1.2k
Forks
169
PR merge metrics
No merged PRs in 30d

Description

hey guys what do you think about giving support to protobuf ?
I have a current need to share/log tensor data via python.

Wanted to discuss here before starting myself.
I image something like this soon in [kornia](https://github.com/kornia/kornia)
```python
from dlpack import dlpack_pb2
import kornia as K

img: K.core.Image = K.io.read_image(...)

# img.to_proto() will convert tensor data (numpy/torch) to `dlpack_pb2`
K.io.save(..., img.to_proto())

img_proto: dlpack_pb2.DLTensor = K.io.load(....)
img_load: K.core.Image = Image.from_proto(img_proto)
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.