PyTorch 自定义model简单示例
PyTorch 自定义model简单示例环境:python3.8Pytorch介绍Pytorch中文文档import torchimport numpy as npdef test(): x_data = torch.Tensor([[2,2,1],[1,1,2],[-1,0,1],[-1,1,-1]]) y_data = torch.Tensor([[5,1],[4,2],[0,0],[-1,-3]]) # 对应关系为x [[1,1], #
原创
2021-02-09 15:23:45 ·
1467 阅读 ·
0 评论