Code

Add squeeze for labeled tensors#1434

Code

Merged

ricardoV94 merged 23 commits intopymc-devs:labeled_tensorspymc-devs/pytensor:labeled_tensorsfrom

AllenDowney:add_expand_dims_squeezeAllenDowney/pytensor:add_expand_dims_squeezeCopy head branch name to clipboard

Jun 6, 2025

Commits

Commits on May 30, 2025

• Add ExpandDims and Squeeze operations with tests and rewrite rules

  AllenDowney

  committed
  b3e859c

  Copy full SHA for b3e859c

• fix: match xarray's expand_dims behavior with None dimension - Update ExpandDims op and rewrite rule to not add a new dimension when dim is None - Update tests to verify behavior matches xarray

  AllenDowney

  committed
  d824870

  Copy full SHA for d824870

• Simplify local_squeeze_reshape function by removing redundant checks and streamlining logic.

  AllenDowney

  committed
  a076966

  Copy full SHA for a076966

• Ruff

  AllenDowney

  committed
  e2ffe1c

  Copy full SHA for e2ffe1c

Commits on Jun 2, 2025

• Restoring the commit with xarray-based tests

  AllenDowney

  committed
  7489489

  Copy full SHA for 7489489

• Merge fork/add_expand_dims_squeeze, keeping our changes

  AllenDowney

  committed
  22adb6f

  Copy full SHA for 22adb6f

• Updating squeeze

  AllenDowney

  committed
  a7e2bf8

  Copy full SHA for a7e2bf8

• Working on expand_dims

  AllenDowney

  committed
  332139d

  Copy full SHA for 332139d

Commits on Jun 3, 2025

• Working on squeeze

  AllenDowney

  committed
  4b2f0f7

  Copy full SHA for 4b2f0f7

Commits on Jun 4, 2025

• Organizing squeeze tests

  AllenDowney

  committed
  2120b1a

  Copy full SHA for 2120b1a

• Working on expand_dims

  AllenDowney

  committed
  2bb1fce

  Copy full SHA for 2bb1fce

• Merge branch 'labeled_tensors' into add_expand_dims_squeeze

  AllenDowney

  authored
  dd13fc7

  Copy full SHA for dd13fc7

• lint

  AllenDowney

  committed
  7a308b9

  Copy full SHA for 7a308b9

Commits on Jun 5, 2025

• Update pytensor/xtensor/shape.py

  Show description for 9c1a0b7
  

  AllenDowney

  and

  ricardoV94

  authored
  9c1a0b7

  Copy full SHA for 9c1a0b7

• Cleaning up squeeze

  AllenDowney

  committed
  1024798

  Copy full SHA for 1024798

• Removing expand_dims

  AllenDowney

  committed
  260b9b6

  Copy full SHA for 260b9b6

• Removing unneded check

  AllenDowney

  committed
  915a368

  Copy full SHA for 915a368

• Update pytensor/xtensor/shape.py

  Show description for 05dac9e
  

  AllenDowney

  and

  ricardoV94

  authored
  05dac9e

  Copy full SHA for 05dac9e

• Update pytensor/xtensor/shape.py

  Show description for 98d297e
  

  AllenDowney

  and

  ricardoV94

  authored
  98d297e

  Copy full SHA for 98d297e

• Update pytensor/xtensor/rewriting/shape.py

  Show description for 3202c4c
  

  AllenDowney

  and

  ricardoV94

  authored
  3202c4c

  Copy full SHA for 3202c4c

• Update pytensor/xtensor/shape.py

  Show description for 8d4fdd5
  

  AllenDowney

  and

  ricardoV94

  authored
  8d4fdd5

  Copy full SHA for 8d4fdd5

• All but one requested change

  AllenDowney

  committed
  f000bbb

  Copy full SHA for f000bbb

• Picking a nit

  AllenDowney

  committed
  2de9566

  Copy full SHA for 2de9566