| 864 | } |
| 865 | |
| 866 | Shape Broadcast::output_shape(const std::vector<array>& inputs) { |
| 867 | auto shape = inputs[0].shape(); |
| 868 | for (int i = 1; i < inputs.size(); ++i) { |
| 869 | shape = broadcast_shapes(shape, inputs[i].shape()); |
| 870 | } |
| 871 | return shape; |
| 872 | } |
| 873 | |
| 874 | std::vector<Shape> Broadcast::output_shapes(const std::vector<array>& inputs) { |
| 875 | if (inputs.size() < 2) { |