ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

jittor和torch的爱恨情仇

2026/8/4 17:35:53 拓冰建站 浏览量
jittor和torch的爱恨情仇

左jittor右torch

execute --- forward
concat --- cat
zeros_like --- empty_like
jt.array --- torch.tensor
greater --- gt

需要注意的
unique在 Jittor 中只有 return_inverse=Truereturn_counts=True 才有效,torch 中只需要 return_counts=True 即可

DataLoader在 Jittor 中是一个函数,而在 torch 中是一个class