在张量流中,使op时,REGISTER_OP中的SetShapeFn和Op(Compute)函数中的区别是什么?

时间:2018-07-20 07:58:05

标签: tensorflow

我正在尝试在Tensorflow中创建新操作,但是我对设置输出形状有一些疑问。

在Compute函数中设置具有某种形状的输出与在REGISTER_OP中设置SetShapeFn有什么区别?

如果在Compute函数中设置不同的形状(与REGISTER_OP中的SetShapeFn相比)是否有任何不利之处?

例如,如果在SetShapeFn中将输出的形状设置为(100,100),是否应该在(Compute)函数中将输出的形状设置为(100,100)? (如何设置(50,100)?有什么区别吗?)

谢谢。

0 个答案:

没有答案