Scalars are 0-dim, not 1-dim (pytorch#239)

bertmaher · Mikhail Zolotukhin · commit dd7d232044c8 · 2020-03-03T15:22:17.000-08:00
diff --git a/torch/csrc/jit/tensorexpr/kernel.cpp b/torch/csrc/jit/tensorexpr/kernel.cpp
@@ -191,7 +191,7 @@ static std::vector<ExprHandle> broadcastShapes(
 std::vector<ExprHandle> TensorExprKernel::valueShape(const torch::jit::Value* v) {
   auto it = tensors_.find(v->unique());
   if (it == tensors_.end()) {
-    return {1};
+    return {};
   }
   return ExprVectorToExprHandleVector(it->second->dims());
 }

Original file line number	Diff line number	Diff line change
`@@ -191,7 +191,7 @@ static std::vector<ExprHandle> broadcastShapes(`
`191`	`191`	`std::vector<ExprHandle> TensorExprKernel::valueShape(const torch::jit::Value* v) {`
`192`	`192`	`auto it = tensors_.find(v->unique());`
`193`	`193`	`if (it == tensors_.end()) {`
`194`		`- return {1};`
	`194`	`+ return {};`
`195`	`195`	`}`
`196`	`196`	`return ExprVectorToExprHandleVector(it->second->dims());`
`197`	`197`	`}`