将多维张量转储到文件

时间:2018-11-01 09:45:04

标签: c++ tensorflow

我有一个多维张量,我想将其作为矩阵转储到文件中。最好的方法是什么?

目前,我对4维张量具有以下要求:

const tensorflow::TensorShape& tensor_shape = tensor.shape();
auto _tensor = tensor.tensor<float, 4>();
for(int i=0; i<tensor_shape.dim_size(0); i++)
{
    for(int j=0; j<tensor_shape.dim_size(1); j++)
    {

        for(int k=0; k<tensor_shape.dim_size(2); k++)
        {
            for(int l=0; l<tensor_shape.dim_size(3); k++)
            {
               std::cout << _tensor(i,j,k,l) << std::endl; //right one
            }
        }
    }
}

但是这非常慢。.是否有更好,更有效的方法来做到这一点?

0 个答案:

没有答案