Skip to content

Add tensor layout helpers#5

Merged
GiraudJules merged 1 commit into
mainfrom
feat/tensor-layout-helpers
May 26, 2026
Merged

Add tensor layout helpers#5
GiraudJules merged 1 commit into
mainfrom
feat/tensor-layout-helpers

Conversation

@GiraudJules

@GiraudJules GiraudJules commented May 26, 2026

Copy link
Copy Markdown
Owner

Summary

Add tensor layout and channel-order helpers in Rust and expose them through the Python NumPy API.

Changes

  • add HWC/CHW conversion helpers
  • add NHWC/NCHW batch conversion helpers
  • add RGB/BGR channel swap helper
  • extend Rust tests and the Python smoke test

Why

These conversions are common in Python CV and DL pipelines and are a good base for the upcoming batched preprocessing path.

Validation

  • cargo fmt --check
  • cargo test
  • cargo check --features python
  • ../../.venv/bin/maturin develop --features python
  • ../../.venv/bin/python scripts/python_smoke_test.py

@GiraudJules GiraudJules self-assigned this May 26, 2026
@GiraudJules GiraudJules marked this pull request as ready for review May 26, 2026 17:20
@GiraudJules GiraudJules merged commit a2d2f79 into main May 26, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant