reshape validation fix

Signed-off-by: raver119 <raver119@gmail.com>
This commit is contained in:
raver119 2019-12-06 21:19:33 +03:00
parent b32dd1bf92
commit 70e08c3a6c

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;