PyTorch是最流行的深度学习框架之一。
张量操作
import torch
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.randn(3, 4)神经网络
import torch.nn as nn
class Net(nn.Module):
def __init__(self):
super().__init__()
self.fc1 = nn.Linear(784, 128)
self.fc2 = nn.Linear(128, 10)
def forward(self, x):
x = torch.relu(self.fc1(x))
return self.fc2(x)
评论 (0)
暂无评论，来发表第一条评论吧！
发表评论