-
Notifications
You must be signed in to change notification settings - Fork 5.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add split and merge lod tensor operator #5537
Conversation
please add an issue to describe the background of this PR~ |
I think Split/Merge_Lod_Tensor are not good because there are many situations that need to split and merge LoDTensorArray in different ways. |
@jacquesqiao Yes, it seems that ConditionalFilterOp/ConditionalMergeOp is more accurate. We actually use a mask of bool column vector to filter a LoDTensor. And the LoDTensor is filtered into two parts, true part and false part. Each part will be sent to a separate block. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Fix #5601