jittor和torch的爱恨情仇

左jittor右torch

execute --- forward
concat --- cat
zeros_like --- empty_like
jt.array --- torch.tensor
greater --- gt
greater_equal -- gt_
scatter --- scatter_add_

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

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

posted @ 2025-12-09 01:09  oisdoaiu  阅读(7)  评论(1)    收藏  举报