Plans for supporting Extension Array to support Fixed shape tensor Array

[haixuanTao]

Is your feature request related to a problem or challenge? Please describe what you are trying to do.

Currently, there is no way easy way to support tensor data, which could be handy to do things like AI. And it seems that there is no tracking issue.

See: https://arrow.apache.org/docs/format/CanonicalExtensions.html
And example python implementation:
See: https://arrow.apache.org/docs/python/generated/pyarrow.ExtensionArray.html

Describe the solution you'd like
Implementation should be pretty straightforward as there already a Tensor Primitive Type.

Additional context
Related issue: #4805

[tustvold]

See #4472 for prior discussion on this topic

[haixuanTao]

Thanks. Did not find it at first.

[JackKelly]

For anyone else who (like me) stumbles across this issue: FixedShapeTensor, and the other canonical extension types, were merged into arrow-rs in version 54.1.0 (Feb 2025).

• Docs for FixedShapeTensor
• PR Add ExtensionType trait and CanonicalExtensionType enum #5822