blob: 4eb6eb4e4da44115bc1889241881b39e8e066fde (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
op {
graph_op_name: "ResourceScatterAdd"
in_arg {
name: "resource"
description: <<END
Should be from a `Variable` node.
END
}
in_arg {
name: "indices"
description: <<END
A tensor of indices into the first dimension of `ref`.
END
}
in_arg {
name: "updates"
description: <<END
A tensor of updated values to add to `ref`.
END
}
summary: "Adds sparse updates to the variable referenced by `resource`."
description: <<END
This operation computes
# Scalar indices
ref[indices, ...] += updates[...]
# Vector indices (for each i)
ref[indices[i], ...] += updates[i, ...]
# High rank indices (for each i, ..., j)
ref[indices[i, ..., j], ...] += updates[i, ..., j, ...]
Duplicate entries are handled correctly: if multiple `indices` reference
the same location, their contributions add.
Requires `updates.shape = indices.shape + ref.shape[1:]` or `updates.shape = []`.
<div style="width:70%; margin:auto; margin-bottom:10px; margin-top:20px;">
<img style="width:100%" src='https://www.tensorflow.org/images/ScatterAdd.png' alt>
</div>
END
}
|
