返回代表張量的 dlpack 膠囊。
用法
tf.experimental.dlpack.to_dlpack(
tf_tensor
)參數
-
tf_tensorTensorFlow Eager tensor，要轉換為 dlpack 膠囊。
返回
- 一個名為 dltensor 的 PyCapsule，它與其他框架共享底層內存。這個 PyCapsule 隻能使用一次。
此操作確保底層數據存儲器在返回時準備就緒。
a = tf.tensor([1, 10])
dlcapsule = tf.experimental.dlpack.to_dlpack(a)
# dlcapsule represents the dlpack data structure相關用法
- Python tf.experimental.dlpack.from_dlpack用法及代碼示例
- Python tf.experimental.dispatch_for_unary_elementwise_apis用法及代碼示例
- Python tf.experimental.dispatch_for_api用法及代碼示例
- Python tf.experimental.dispatch_for_binary_elementwise_apis用法及代碼示例
- Python tf.experimental.numpy.iinfo用法及代碼示例
- Python tf.experimental.Optional.has_value用法及代碼示例
- Python tf.experimental.unregister_dispatch_for用法及代碼示例
- Python tf.experimental.tensorrt.Converter用法及代碼示例
- Python tf.experimental.ExtensionType用法及代碼示例
- Python tf.experimental.Optional.get_value用法及代碼示例
- Python tf.experimental.numpy.issubdtype用法及代碼示例
- Python tf.experimental.numpy.unicode_用法及代碼示例
- Python tf.experimental.async_scope用法及代碼示例
- Python tf.experimental.Optional.empty用法及代碼示例
- Python tf.experimental.numpy.float16.as_integer_ratio用法及代碼示例
- Python tf.experimental.numpy.float64.as_integer_ratio用法及代碼示例
注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.experimental.dlpack.to_dlpack。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。