Skip to content

Commit 8422fcb

Browse files
author
yifan_shen3
committed
address review comment on scatter_nd doc: indices and updates have different shapes; make doc more explicit
1 parent d60574c commit 8422fcb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

coremltools/converters/mil/mil/ops/defs/iOS15/scatter_gather.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -512,9 +512,9 @@ class scatter_nd(Operation):
512512
----------
513513
data: tensor<\*D, T> (Required)
514514
indices: tensor<\*E, i32> (Required)
515-
* E[-1] <= len(D)
516-
updates: tensor<\*E, T> (Required)
517-
* Must be the shape as ``E[:-1] + data.shape[E[-1]:]``.
515+
* indices.shape[-1] <= data.rank
516+
updates: tensor<\*F, T> (Required)
517+
* Must be the shape as ``indices.shape[:-1] + data.shape[indices.shape[-1]:]``.
518518
mode: const string (Optional)
519519
* Default to ``add``.
520520
* Can be the following modes: ``update``, ``add``, ``sub``, ``mul``,

0 commit comments

Comments
 (0)