返回代表张量的 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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。