Tensorflow C ++,如何将tensorflow :: Output转换为Tensor

时间:2018-09-13 07:20:44

标签: c++ tensorflow

Tensorflow C ++,如何在不使用tensorflow::Output函数的情况下将类型Tensor转换为类型Run()

Conv2D()之后,我想使用FusedBatchNorm()的功能。

auto Conv2DOut = Conv2D(...);

auto BatchNorm = FusedBatchNorm(..., Conv2DOut, ...);

但是函数FusedBatchNorm()的输入参数x的类型是Tensor,而不是tensorflow::Output

因此,我需要将Conv2DOut转换为Tensor。如果无法完成,则可能是TensorFlow C ++的错误。

0 个答案:

没有答案