- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for ReduceSum (0.31 sec)
-
tensorflow/cc/gradients/array_grad.cc
// all the gradients from the shape it fills. // We use ReduceSum to implement this, which needs an argument providing // the indices of all the dimensions of the incoming gradient. // grad(x) = reduce_sum(grad(y), [0..rank(grad(y))]) auto all_dims = Range(scope, Const(scope, 0), Rank(scope, grad_inputs[0]), Const(scope, 1)); grad_outputs->push_back(ReduceSum(scope, grad_inputs[0], all_dims)); return scope.status();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0)