pytorch code

1.二分类问题代码
2.线性回归问题代码
3.多头注意力机制代码