diff options
author | Adam Paszke <adam.paszke@gmail.com> | 2016-11-23 15:26:16 +0300 |
---|---|---|
committer | Adam Paszke <adam.paszke@gmail.com> | 2016-11-23 15:26:16 +0300 |
commit | 15098c49a9e2ebda619e34992a329232097cfcd6 (patch) | |
tree | 17de6c74854595fe58c3b2347187c1d2be588ddc | |
parent | d2da5012374ea4b8e3556f3ebe0ca00173df0a50 (diff) |
Use TH_INDEX_BASE for range asserts in MultiLabelMarginCriterion
-rw-r--r-- | lib/THNN/generic/MultiLabelMarginCriterion.c | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/THNN/generic/MultiLabelMarginCriterion.c b/lib/THNN/generic/MultiLabelMarginCriterion.c index fe851c9..16398c1 100644 --- a/lib/THNN/generic/MultiLabelMarginCriterion.c +++ b/lib/THNN/generic/MultiLabelMarginCriterion.c @@ -35,8 +35,8 @@ void THNN_(MultiLabelMarginCriterion_updateOutput)( && (target->size[1] == dim), 3, "inconsistent target size"); } - THArgCheck(THIndexTensor_(minall)(target) >= 0, 3, "target out of range"); - THArgCheck(THIndexTensor_(maxall)(target) <= dim, 3, "target out of range"); + THArgCheck(THIndexTensor_(minall)(target) >= -1+TH_INDEX_BASE, 3, "target out of range"); + THArgCheck(THIndexTensor_(maxall)(target) < dim+TH_INDEX_BASE, 3, "target out of range"); target = THIndexTensor_(newContiguous)(target); input = THTensor_(newContiguous)(input); @@ -128,8 +128,8 @@ void THNN_(MultiLabelMarginCriterion_updateGradInput)( && (isTarget->size[1] == dim), 3, "inconsistent isTarget size"); } - THArgCheck(THIndexTensor_(minall)(target) >= 0, 3, "target out of range"); - THArgCheck(THIndexTensor_(maxall)(target) <= dim, 3, "target out of range"); + THArgCheck(THIndexTensor_(minall)(target) >= -1+TH_INDEX_BASE, 3, "target out of range"); + THArgCheck(THIndexTensor_(maxall)(target) < dim+TH_INDEX_BASE, 3, "target out of range"); THArgCheck(THTensor_(minall)(isTarget) >= 0, 3, "isTarget out of range"); THArgCheck(THTensor_(maxall)(isTarget) <= 1, 3, "isTarget out of range"); |