gpt4 book ai didi

python - pytorch中张量的df.mask之类的类似功能吗?

转载 作者:太空宇宙 更新时间:2023-11-04 09:23:14 26 4
gpt4 key购买 nike

我想用 -5 替换二维张量中的所有 0。

有了数据框,我可以很容易地做到这一点:

df = df.mask(df=0, -5) 

但这不适用于张量。我试过:

 y = torch.where(y = 0, -5, y) 

最佳答案

很简单,就用这个

y[y==0]=-5

关于python - pytorch中张量的df.mask之类的类似功能吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59037704/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com