我正在尝试在Tensorflow中创建新操作,但是我对设置输出形状有一些疑问。
在Compute函数中设置具有某种形状的输出与在REGISTER_OP中设置SetShapeFn有什么区别?
如果在Compute函数中设置不同的形状(与REGISTER_OP中的SetShapeFn相比)是否有任何不利之处?
例如,如果在SetShapeFn中将输出的形状设置为(100,100),是否应该在(Compute)函数中将输出的形状设置为(100,100)? (如何设置(50,100)?有什么区别吗?)
谢谢。