这是一个简单的custom torch 由于存在很多bug,只能作为CNN的辅助理解(不保证正确),认真你就输了。 不过之后也可能会做一些修改,也有可能哪天就变成read only了 具体调用方法和pytorch相似,就是可用功能很少,极简化框架(确信), bug也很多,没准你随便跑一下就出bug了 参考文章: microsoft ai softmax 反向传播 子类化numpy