BroadcastGradientArgs没有提供文档

Question

我正在创建自定义操作。在检查后支柱中的操作时，我遇到了BroadcastGradientArgs。

有没有人知道这是做什么的？

Answer 1

它是一个内部op，它返回给定两个张量形状的缩减轴。请注意，它的返回值始终用于reduce_sum。支持广播的操作（涉及较小等级或形状的张量的操作）需要具有缩减功能，以使得到的梯度具有相同的原始大小。它具有将各个梯度相加为一个值的效果。