pytorch和jittor 程序对比

必须在这里说明后期本人进行了gpu上的测试相比cpu上jittor落后的这一点点
gpu上jittor 性能领先的不是一点点是4-5倍
https://dongfangyou.blog.csdn.net/article/details/111461043
jittor代码

import jittor as jt
import numpy as np
from jittor import nn, Module, init
### model define
import time

class Model(Module