def backward(self, E):
// TODO FIXME this is test only
return E
class Aggregate(Module):
def forward(self, A, R):
After Change
def backward(self, gradE):
A, R = self.saved_tensors
gradA = A.clone()
gradR = R.clone()
encoding_lib.Encoding_Float_aggregate_backward(gradA, gradR, gradE,
A, R)
return gradA, gradR