Tensorflow C API使用的底层数据缓冲区的内存布局是什么?

时间:2018-02-02 20:49:10

标签: c tensorflow

我正在使用TF_TensorData来读取我的图形的输出,这是一个尺寸{1,520,420,1}的张量。我的目标是拉出一张520 x 420的花车图像。我想知道用TF_TensorData检索的底层数据缓冲区的布局是什么。在输入端,假设传递给TF_NewTensor的数据缓冲区的布局是相同的,我是否正确?

在一般情况下,任意尺寸张量的布局是什么?

2 个答案:

答案 0 :(得分:0)

{1,520,420,1}与{520,420}

的形状相同
  

在输入端,假设传递给TF_NewTensor的数据缓冲区的布局是否相同,我是否正确?

必须相同,只需导入一些数据和打印输出。

答案 1 :(得分:0)

对于你的两个张量,是的布局是相同的。

根据tensorflow/c/c_api.h中的文档,数据按行主要顺序排列。

相关问题