繁体   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