# case1
import paddle

paddle.enable_static()
x = paddle.static.data(name='x', shape=[-1, 1, -1], dtype='float32')
index = paddle.static.data(name='index', shape=[-1, 1, -1], dtype='bool')
out = x[index]

# case2
paddle.enable_static()
x = paddle.static.data(name="x", shape=[-1, -1, -1])
index = paddle.static.data(name="index", shape=[-1, 1, -1], dtype='bool')
a = paddle.static.data(name="a", shape=[-1, ])
# breakpoint()
x[index] = a
print(x)