簡體   English   中英

Tensorflow C API使用的基礎數據緩沖區的內存布局是什么?

[英]What is the memory layout of the underlying data buffer used by the Tensorflow C API?

我正在使用TF_TensorData從圖形中讀取輸出,該輸出是尺寸為{1,520,420,1}的張量。 我的目標是提取520 x 420的彩車圖像。 我想知道用TF_TensorData檢索的基礎數據緩沖區的布局是什么。 在輸入方面,我假設傳遞給TF_NewTensor的數據緩沖區的布局相同是正確的嗎?

在一般情況下,任意維的張量的布局是什么?

對於您的兩個張量,是的,布局是相同的。

根據tensorflow/c/c_api.h的文檔,數據按行優先順序排列。

{1,520,420,1}與{520,420}的形狀相同

在輸入方面,我假設傳遞給TF_NewTensor的數據緩沖區的布局相同是正確的嗎?

必須相同,只是導入一些數據並打印輸出。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM