Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
https://cg.cs.tsinghua.edu.cn/jittor/
Apache License 2.0
3.08k stars 311 forks source link

where()函数与torch行为不一致 #308

Open wzever opened 2 years ago

wzever commented 2 years ago

Var.where()方法与torch中的行为似乎不一致,请问有实现类似功能的函数吗?

torch.where(condition, x, y) -> [Tensor] Return a tensor of elements selected from either x or y, depending on condition.

屏幕截图 2022-04-04 234532
cjld commented 2 years ago

谢谢您的反馈,最新版本jittor1.3.2.3已经与torch保持一致,目前jt.where同时支持一个参数和三个参数,其中一个参数的版本和numpy保持一致,三个参数的版本和torch保持一致

wzever commented 2 years ago

谢谢回复。但在pip更新的时候报错: UnicodeDecodeError: 'utf-8' codec can't decode byte 0xb7 in position 8110: invalid start byte in installed-files.txt file at path: c:\users\...\python\python39\lib\site-packages\jittor-1.3.1.55-py3.9.egg-info\installed-files.txt

屏幕截图 2022-04-05 135241

请问该如何解决?感谢!

Jittor commented 2 years ago

您试试删除然后重新安装呢?

---原始邮件--- 发件人: "Pan @.> 发送时间: 2022年4月5日(周二) 中午1:57 收件人: @.>; 抄送: @.***>; 主题: Re: [Jittor/jittor] where()函数与torch行为不一致 (Issue #308)

谢谢回复。但在pip更新的时候报错: UnicodeDecodeError: 'utf-8' codec can't decode byte 0xb7 in position 8110: invalid start byte in installed-files.txt file at path: c:\users...\python\python39\lib\site-packages\jittor-1.3.1.55-py3.9.egg-info\installed-files.txt

请问该如何解决?感谢!

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>