Merge 9d1108a into a16c5b5

gorgonia · Sep 28, 2020 · 80ed80b · 80ed80b
2 parents a16c5b5 + 9d1108a
commit 80ed80b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/op_tensor.go b/op_tensor.go
@@ -463,6 +463,9 @@ func (op repeatOp) UsePreallocDo(prealloc Value, inputs ...Value) (retVal Value,
 		err = errors.Errorf(nyiTypeFail, "repeatOp.Do()", inputs[0])
 		return
 	}
+	if rep == 1 {
+		return Copy(pt, t)
+	}
 
 	return tensor.RepeatReuse(t, pt, op.along, rep)
 }