PyTorch 自定义model简单示例
PyTorch 自定义model简单示例
环境:python3.8
Pytorch介绍
Pytorch中文文档
import torch
import numpy as np
def 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 ·
1431 阅读 ·
0 评论