作为其中的一部分,我必须将维数a x b x c的三维张量转换为维数a x b*c的二维张量。Arbitrary 3D tensornet:add(nn.View(-1):setNumInputDims(2))现在,我想修改我的网络因此,我想把维数d x a x b x c的四维张量转化为维数d x a x b*c的三维张量,其中d是我的小批中的元素数。d是事先已知的,但其他维度则不然。当我把一个四维张量</em
Use Eager execution or decorate this function with @tf.function.我确实知道我必须以tf.cond()的形式重写if条件。然而,我不知道如何在张量流中编写break,我也不确定是哪个if条件导致了问题,因为这个笔记本中的相同函数是否工作正常?有任何帮助吗?