Add broadcast #231

Merged
AlyaNovikova merged 3 commits from feature/tensor-algebra into feature/tensor-algebra 2021-03-12 18:11:55 +03:00
Showing only changes of commit 626d5c98fa - Show all commits

View File

@ -75,7 +75,7 @@ public class RealTensorAlgebra : TensorPartialDivisionAlgebra<Double, RealTensor
val curDim = shape[i] val curDim = shape[i]
val offset = totalDim - shape.size val offset = totalDim - shape.size
if (curDim != 1 && totalShape[i + offset] != curDim) { if (curDim != 1 && totalShape[i + offset] != curDim) {
throw IllegalArgumentException("Shapes are not compatible and cannot be broadcast") throw RuntimeException("Shapes are not compatible and cannot be broadcast")
} }
} }
} }