/usr/local/lib64/python3.6/site-packages/torch/include/c10/core
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/c10/core/WrapDimMinimal.h (805B)
#pragma once
#include
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;
}
} // namespace c10