PaddlePaddle的布尔索引如何实现?
收藏
下列代码是Pytorch实现的多维张量索引和赋值操作。但是相应的paddle则会报错。请问paddle有什么类似的操作吗吗?
>>> x = torch.randn((10, 2))
>>> x
tensor([[ 0.2217, -0.6653],
[-0.7827, 0.7083],
[-1.0004, 0.5266],
[-1.2800, 2.1070],
[ 0.3620, -0.4531],
[ 1.2241, 0.5158],
[ 0.1879, -0.2660],
[-0.8315, -0.1696],
[ 0.3561, -1.4320],
[ 0.0403, -0.7668]])
>>> x[x[:, 0]<0] = 0
>>> x
tensor([[ 0.2217, -0.6653],
[ 0.0000, 0.0000],
[ 0.0000, 0.0000],
[ 0.0000, 0.0000],
[ 0.3620, -0.4531],
[ 1.2241, 0.5158],
[ 0.1879, -0.2660],
[ 0.0000, 0.0000],
[ 0.3561, -1.4320],
[ 0.0403, -0.7668]])
0
收藏
请登录后评论
这个建议issue,可能是使用方法不一样