Code

Simplify subtensor shape inference#1299

Code

Draft

ricardoV94 wants to merge 7 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:simplify_subtensor_shapericardoV94/pytensor:simplify_subtensor_shapeCopy head branch name to clipboard

Commits

Commits on Mar 25, 2025

• Remove Scalar prefix from Maximum, Minimum, and Softplus

  ricardoV94

  committed
  a56e3a5

  Copy full SHA for a56e3a5

• Simplify graph returned by Subtensor.infer_shape

  ricardoV94

  committed
  0dc2a1e

  Copy full SHA for 0dc2a1e

• Do not bother with `nan` in discrete Maximum/Minimum

  ricardoV94

  committed
  34ebed5

  Copy full SHA for 34ebed5

• Make Maximum and Minimum variadic

  ricardoV94

  committed
  2804e47

  Copy full SHA for 2804e47

• Implement MinMax algebra rewrites to simplify slice shape graphs

  ricardoV94

  committed
  a935c01

  Copy full SHA for a935c01

• .

  ricardoV94

  committed
  aaecd79

  Copy full SHA for aaecd79

• Don't apply `local_upcast_elemwise_constant_inputs` when all inputs are constant

  ricardoV94

  committed
  9a9fbbc

  Copy full SHA for 9a9fbbc