Tensorflow C ++,如何在不使用tensorflow::Output
函数的情况下将类型Tensor
转换为类型Run()
。
在Conv2D()
之后,我想使用FusedBatchNorm()
的功能。
auto Conv2DOut = Conv2D(...);
auto BatchNorm = FusedBatchNorm(..., Conv2DOut, ...);
但是函数FusedBatchNorm()
的输入参数x的类型是Tensor
,而不是tensorflow::Output
。
因此,我需要将Conv2DOut
转换为Tensor
。如果无法完成,则可能是TensorFlow C ++的错误。