array-api icon indicating copy to clipboard operation
array-api copied to clipboard

RFC: add support for complex input to `mean`

Open lucascolley opened this issue 1 year ago • 2 comments

x-ref https://github.com/data-apis/array-api-extra/pull/10/files#r1776010210

While implementing a simple cov in array-api-extra (https://github.com/data-apis/array-api-extra/pull/10) I found that xp.mean does not support complex input. @kgryte suggested opening an issue over here as the arithmetic mean is well-defined for complex numbers.

lucascolley avatar Sep 25 '24 21:09 lucascolley

Was there a reason for this or was it just an accidental omission? Mean is well-defined for complex numbers (https://en.wikipedia.org/wiki/Complex_random_variable), and should be easy to implement since you either just apply the real kernel formula directly to complex data types, or take the means of the real and imaginary parts separately.

asmeurer avatar Sep 25 '24 22:09 asmeurer

I believe an omission and thus far no R&D whether supported across all array libraries.

kgryte avatar Sep 25 '24 23:09 kgryte

I've opened https://github.com/data-apis/array-api/pull/850.

kgryte avatar Oct 31 '24 09:10 kgryte