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) {
TORCH_CHECK_INDEX(false, "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) {
TORCH_CHECK_INDEX(false,
"Dimension out of range (expected to be in range of [",
min, ", ", max, "], but got ", dim, ")");
}
if (dim < 0) dim += dim_post_expr;
return dim;
}
}
|
