我有一个多维张量,我想将其作为矩阵转储到文件中。最好的方法是什么?
目前,我对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
}
}
}
}
但是这非常慢。.是否有更好,更有效的方法来做到这一点?