Нейронная сеть в pytorch

Question

Я хочу создать нейронную сеть в PyTorch, которая будет иметь 2 входа и 3 выхода с 1 скрытым слоем. Два входа будут числами с плавающей запятой, которые представляют особенности изображения, а 3 выхода будут действительными числами от 0 до 1. Например, выход (1, 0, 0) будет означать, что он квадратный, а (0,1,0) будет означает, что это прямоугольник. Есть идеи, как это сделать в pytorch?

Answer 1

Сеть может быть определена следующим образом:

import torch
import torch.nn as nn
import torch.optim as optim
import torch.autograd as autograd 
import torch.nn.functional as F
from torch.autograd import Variable

class Net(nn.Module):
    def __init__(self, num_inputs=2, num_outputs=3,hidden_dim=5):
        # define your network here
        super(Net, self).__init__()
        self.layer1 = nn.Linear(num_inputs,hidden_dim)
        self.layer2 = nn.Linear(hidden_dim,num_outputs)


    def forward(self, x):
        # implement the forward pass
        x = F.relu(self.layer1(x))
        x = F.sigmoid(self.layer2(x))
        return x

Хотя я определил здесь сеть, вам, возможно, стоит взглянуть на некоторые примеры на официальном веб-сайте pytorch, например, о том, как обучать вашу модель.