diff options
| -rw-r--r-- | chapters/data_layout.adoc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/chapters/data_layout.adoc b/chapters/data_layout.adoc index 1c87d4b..205ccaf 100644 --- a/chapters/data_layout.adoc +++ b/chapters/data_layout.adoc @@ -293,7 +293,7 @@ Permutes the dimensions based on perm. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor with rank from 1 to 4 +|Input|in_t*|input1|shape1|Input tensor with minimum rank of one. |Attribute|int32_t|perms|[rank(input1)]|List of integers of length equal to the rank of input1. Values must be valid dimensions within shape1, and may not be repeated. |Output|in_t*|output|shape|Output tensor of same type, rank as the input tensor |=== @@ -304,7 +304,7 @@ Permutes the dimensions based on perm. ---- for_each(index in perms) { // Ensure each perms value is a valid value - ERROR_IF(index > rank(shape1)); + ERROR_IF(index >= rank(shape1)); ERROR_IF(index < 0); // Ensure ranks aren't repeated ERROR_IF(indexes_used[index] == true); |