Commit 317c5086 authored by Isaak Lim's avatar Isaak Lim
Browse files

remove overly strict assertion on features dimensionality since

features will be an empty torch.FloatTensor when first calling set_locations
parent cb129912
...@@ -58,7 +58,6 @@ extern "C" void scn_D_(setInputSpatialLocations)(void **m, ...@@ -58,7 +58,6 @@ extern "C" void scn_D_(setInputSpatialLocations)(void **m,
THLongTensor *locations, THLongTensor *locations,
THFloatTensor *vecs, THFloatTensor *vecs,
bool overwrite) { bool overwrite) {
assert(features->nDimension == 2 and "features must be 2 dimensional!");
assert(locations->nDimension == 2 and "locations must be 2 dimensional!"); assert(locations->nDimension == 2 and "locations must be 2 dimensional!");
assert(vecs->nDimension == 2 and "vecs must be 2 dimensional!"); assert(vecs->nDimension == 2 and "vecs must be 2 dimensional!");
assert(locations->size[0] == vecs->size[0] and assert(locations->size[0] == vecs->size[0] and
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment