[Add] broadcasting for Arithmetic Operators
This merge request is dedicated to enabling broadcasting on binary operators (Add, Mul, Div, Sub, Pow). The expected behaviour is the same as numpy
Here are examples of supported broadcasting
Tensor_A shape | 3 | 2 | 1 | 3 |
---|---|---|---|---|
Tensor_B shape | 1 | 2 | 4 | 3 |
Output shape | 3 | 2 | 4 | 3 |
Tensor_A shape | 3 | 2 | 1 | 3 |
---|---|---|---|---|
Tensor_C shape | 3 | |||
Output shape | 3 | 2 | 1 | 3 |
Edited by Maxence Naud