ValueError: For 'MaskedSelect', the two input 'input' and 'mask' with shape: [const vector][10] and [const vector][8] can not broadcast.

import mindspore as ms
import mindspore.ops as ops
import mindspore.nn as nn
import numpy as np
from mindspore import Tensor

class TestNet(nn.Cell):
    def construct(self, x, y):
        mask = y > 1
        x = ops.masked_select(x, mask[:, None]).reshape(-1, 5)
        y = ops.masked_select(y, mask)

        mask = y > 3
        x = ops.masked_select(x, mask[:, None]).reshape(-1, 5)
        y = ops.masked_select(y, mask)

        mask = y > 5
        x = ops.masked_select(x, mask[:, None]).reshape(-1, 5)
        return x

def main():
    ms.set_context(mode=ms.GRAPH_MODE)
    net = TestNet()

    x = Tensor(np.arange(50).reshape(10, 5))
    y = Tensor(np.arange(10))

    result = net(x, y)
    assert result.shape == (4, 5)

if __name__ == '__main__':
    main()