您好，请问在一些需要计算多个输入图像转换副本的平均梯度时，代码里是否有实现计算平均梯度进行扰动更新这一操作？

[lazylittlezhao]

比如SIA论文里描述的需要计算平均梯度，但是我在代码中没有发现这一步骤的显示实现，我代码能力比较薄弱qwq，希望可以解答我的困惑，十分感谢。

` for _ in range(self.epoch):
# Obtain the output
logits = self.get_logits(self.transform(data+delta, momentum=momentum))

        # Calculate the loss
        loss = self.get_loss(logits, label)

        # Calculate the gradients
        grad = self.get_grad(loss, delta)

        # Calculate the momentum
        momentum = self.get_momentum(grad, momentum)

        # Update adversarial perturbation
        delta = self.update_delta(delta, data, momentum, self.alpha)

    return delta.detach()`