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

View File

@ -164,6 +164,7 @@ namespace nd4j {
// we can launch op using Int arguments
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();
int e = 1;