Skip to content

[Bug] [Relay][MXNet] wrong conversion logic in Repeat operator about axis #15890

@jikechao

Description

@jikechao

Bug description

According to the MXNet documentation, the default value of axis is None rather than 0. They are different.
image

None: uses the flattened input array and return a flat output array.
0: repeats the data along the first axis.


For example:

x = [[ 1, 2], [ 3, 4]]
repeat(x, repeats=2) = [ 1., 1., 2., 2., 3., 3., 4., 4.]
repeat(x, repeats=2, axis=0) = [[ 1., 2.], [ 1., 2.], [ 3., 4.], [ 3., 4.]]

cc @shingjan

Metadata

Metadata

Assignees

No one assigned

    Labels

    needs-triagePRs or issues that need to be investigated by maintainers to find the right assignees to address ittype: bug

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions