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,可能是使用方法不一样