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

import jittor as jtimport numpy as npfrom jittor import nn, Module, init### model defineimport timeclass Model(Module):
    def __init__(