Squares each element of the input array, and returns the sum(s) of that. Parameters ---------- a : array_like Input array. axis : int or None, optional Axis along which to calculate. Default is 0. If None, compute over the whole array `a`. Returns
(a, axis=0)
| 474 | |
| 475 | |
| 476 | def _sum_of_squares(a, axis=0): |
| 477 | """ |
| 478 | Squares each element of the input array, and returns the sum(s) of that. |
| 479 | Parameters |
| 480 | ---------- |
| 481 | a : array_like |
| 482 | Input array. |
| 483 | axis : int or None, optional |
| 484 | Axis along which to calculate. Default is 0. If None, compute over |
| 485 | the whole array `a`. |
| 486 | Returns |
| 487 | ------- |
| 488 | sum_of_squares : ndarray |
| 489 | The sum along the given axis for (a**2). |
| 490 | See also |
| 491 | -------- |
| 492 | _square_of_sums : The square(s) of the sum(s) (the opposite of |
| 493 | `_sum_of_squares`). |
| 494 | """ |
| 495 | return da.sum(a * a, axis) |
| 496 | |
| 497 | |
| 498 | def _square_of_sums(a, axis=0): |