-
Notifications
You must be signed in to change notification settings - Fork 3.5k
Closed
Labels
apiissues related to all other APIs: C, C++, Python, etc.issues related to all other APIs: C, C++, Python, etc.
Description
When users get the ort values and do stuff to them, what the experience will be: Currently they will call .numpy() and ortvalue_from_numpy() and some io binding if the tensor is on cuda. With __dlpack__
users can simply call torch.from_dlpack(ort_value)
to get the cuda tensor in pytorch (and similar methods in other frameworks). No io binding setups needed.
Related: #15963
cc @yuslepukhin
Metadata
Metadata
Assignees
Labels
apiissues related to all other APIs: C, C++, Python, etc.issues related to all other APIs: C, C++, Python, etc.