當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.experimental.dlpack.from_dlpack用法及代碼示例


返回 Tensorflow 即刻張量。

用法

tf.experimental.dlpack.from_dlpack(
    dlcapsule
)

參數

  • dlcapsule 一個名為 dltensor 的 PyCapsule

返回

  • 一個 Tensorflow 即刻張量

返回的張量使用來自其他框架的 dlpack 膠囊共享的內存。

a = tf.experimental.dlpack.from_dlpack(dlcapsule)
  # `a` uses the memory shared by dlpack

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.experimental.dlpack.from_dlpack。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。