当我们要实现自己的一些idea时，torch自带的模块和函数已经不能满足，我们需要自己实现层（或者类），一般的做法是把自定义层加入到已有的torch模块中。