Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reshape validation fix 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-12-06 21:19:33 +03:00
parent b32dd1bf92
commit 70e08c3a6c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libnd4j/include/ops/declarable/generic/shape/reshape.cpp
View File

@ -164,6 +164,7 @@ namespace nd4j {
// we can launch op using Int arguments // we can launch op using Int arguments
if (inputShape->size() == 1) { if (inputShape->size() == 1) {
REQUIRE_TRUE(block.numI() > 0, 0, "Reshape: new shape should be provided as NDArray or int arguments, but nothing was defined");
std::vector<int> *arguments = block.getIArguments(); std::vector<int> *arguments = block.getIArguments();
int e = 1; int e = 1;