首页 Paddle框架 帖子详情
PaddlePaddle的布尔索引如何实现?
收藏
快速回复
Paddle框架 问答模型训练 705 1
PaddlePaddle的布尔索引如何实现?
收藏
快速回复
Paddle框架 问答模型训练 705 1

下列代码是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
收藏
回复
全部评论(1)
时间顺序
三岁
#2 回复于2021-10

这个建议issue,可能是使用方法不一样

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户