1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
| # Copyright (c) Facebook, Inc. and its affiliates.
| #
| # This source code is licensed under the MIT license found in the
| # LICENSE file in the root directory of this source tree.
|
| import torch
|
|
| class GradMultiply(torch.autograd.Function):
| @staticmethod
| def forward(ctx, x, scale):
| ctx.scale = scale
| res = x.new(x)
| return res
|
| @staticmethod
| def backward(ctx, grad):
| return grad * ctx.scale, None
|
|
