Skip to content

Generalize local_reduce_join#1709

Draft
ricardoV94 wants to merge 1 commit intopymc-devs:mainfrom
ricardoV94:generalize_reduce_join
Draft

Generalize local_reduce_join#1709
ricardoV94 wants to merge 1 commit intopymc-devs:mainfrom
ricardoV94:generalize_reduce_join

Conversation

@ricardoV94
Copy link
Member

  1. To cases where tensors are not-broadcastable along join axis
  2. To cases where non-join axis are also removed

This requires reintroducing reduction downstream of Join. When this reduction is useless local_reduce_broadcastable will handle them next

Related to #59

1. To cases where tensors are not-broadcastable along join axis
1. To cases where non-join axis are also removed

This requires reintroducing reduction downstream of Join. When this reduction is useless local_reduce_broadcastable will handle them next
@ricardoV94 ricardoV94 marked this pull request as draft October 30, 2025 14:41
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