如何獲得指針型張量的形狀？

Question

我感興趣的是得到它定義爲指針的輸出張量的值：

Tensor* out = nullptr;

我試圖out->get_shape()會拋出錯誤：

error: 'class tensorflow::Tensor' has no member named 'get_shape'; did you mean 'set_shape'? 

如何獲得形狀？

Answer 1

可以使用'shape()'溫控功能訪問張的形狀，它會返回一個TensorShape，打印的值，你可以使用.vec()方法

const TensorShape& out_shape = out.shape(); 
std::cout << "out = " << out_shape.dim_sizes() << std::endl;