Tensorflow.js：op最大值的梯度误差。输入'$ a'的渐变形状为'32,200'，与输入'32,1'的形状不匹配

作者: NetworkAttachedStorage
发布时间: 2024-09-22 09:44:59 (15天前)
转自：

2 条回复

0#
回复此人
故人 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 好吧，我花了一些时间在这上面，看起来这是Tensforflow.js实现中的一个错误。 </p> <P> 如果您遇到同样的问题，可以通过自己应用以下补丁来修复它（我确定tfjs-layers维护者最终会合并这个拉取请求，所以希望您将来不会再遇到这个问题）。 </p> <BLOCKQUOTE> <P> <a href="https://github.com/tensorflow/tfjs-layers/pull/499" rel="nofollow noreferrer"> https://github.com/tensorflow/tfjs-layers/pull/499 </A> </p> </BLOCKQUOTE> <pre> <code> | export function l2Normalize(x: Tensor, axis?: number): Tensor { | return tidy(() => { | const squareSum = tfc.sum(K.square(x), axis, true); - const epsilonTensor = tfc.mul(scalar(epsilon()), tfc.onesLike(x)); + const epsilonTensor = tfc.mul(scalar(epsilon()), tfc.onesLike(squareSum)); | const norm = tfc.sqrt(tfc.maximum(squareSum, epsilonTensor)); | return tfc.div(x, norm); | }); | } </code> </pre> </DIV>

编辑

登录后才能参与评论