如何打印或计算张量

How to print or cout a tensor?

本文关键字:计算 张量 打印 何打印      更新时间:2023-10-16

我想打印一个张量用于调试

tensorflow::Tensor image_tensor;

I tried

std::cout << &image_tensor;

但是我得到了这样的东西:

0x16fd81cf8I

您必须使用。vec或。flat方法。假设张量是int8

// for i in image_tensor.size …
cout << i << “ “ << image_tensor.vec<int8>()(i)

参见此回答如何在c++中填充张量

对于std::cout << &image_tensor;,您要求打印张量的地址-这就是您得到的。

Try std::cout << image_tensor; (without &)