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
| #pragma once
|
| #include <c10/util/Exception.h>
|
| namespace c10 {
|
| static inline int64_t maybe_wrap_dim(int64_t dim, int64_t dim_post_expr, bool wrap_scalar=true) {
| if (dim_post_expr <= 0) {
| if (!wrap_scalar) {
| AT_INDEX_ERROR("dimension specified as ", dim, " but tensor has no dimensions");
| }
| dim_post_expr = 1; // this will make range [-1, 0]
| }
|
| int64_t min = -dim_post_expr;
| int64_t max = dim_post_expr - 1;
| if (dim < min || dim > max) {
| AT_INDEX_ERROR(
| "Dimension out of range (expected to be in range of [",
| min, ", ", max, "], but got ", dim, ")");
| }
| if (dim < 0) dim += dim_post_expr;
| return dim;
| }
|
| }
|
|
