-
Notifications
You must be signed in to change notification settings - Fork 2.4k
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
[PyOV] Extend Python API with Squeeze-15
#27281
Conversation
|
||
|
||
@pytest.mark.parametrize(("input_shape", "axes", "allow_axis_skip", "expected_shape"), [ | ||
((1, 2, 1, 3, 1, 1), [2, 4], True, [1, 2, 3, 1]), |
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.
what will be if axis corresponds to static dimension not equal to 1
for allow_axis_skip
True and False
?
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.
In that case the dimension will remain unchanged, I just checked on master. @barnasm1 could you please confirm? I also modified the test to cover this case.
Co-authored-by: Roman Kazantsev <[email protected]>
Co-authored-by: Roman Kazantsev <[email protected]>
…into squeeze_py Signed-off-by: p-wysocki <[email protected]>
…to squeeze_py Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Details:
Tickets: