com.microsoft - GatherGrad#
GatherGrad - 1 (com.microsoft)#
Version
name: GatherGrad (GitHub)
domain: com.microsoft
since_version: 1
function:
support_level:
shape inference:
This version of the operator has been available since version 1 of domain com.microsoft.
Summary
Attributes
axis: Which axis to gather on. Negative value means counting dimensions from the back. Accepted range in [-r, r-1] Default value is
?
.
Inputs
shape (heterogeneous) - I: Shape of the Gather input X.
indices (heterogeneous) - Tind: Tensor of int32/int64 indices, of any rank q.
dY (heterogeneous) - T: Gradient of output
Outputs
dX (heterogeneous) - T: Gradient of input
Examples