tf c++获取打印输出tensor的值

假如 predict 是你要获取的tensor的值,其类型为 tensorflow::Tensor

auto features_ptr = predict.tensor<float, 4>();
auto N = (int) predict.dim_size(0);
auto H = (int) predict.dim_size(1);
auto W = (int) predict.dim_size(2);
auto C = (int) predict.dim_size(3);

for (int n = 0; n < N; ++n) {
		for (int h = 0; h < H; ++h) {
				for (int w = 0; w < W; ++w) {
						for (int c = 0; c < C; ++c) {
                				std::cout << features_ptr(n, h, w, c) << std::endl;
                		}
                }
         }
 }

以此类推,其他类似。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值