Quais são as vantagens e comodidades do PyTorch?

O PyTorch oferece várias vantagens principais que o tornam conveniente para desenvolvimento e pesquisa em aprendizagem de máquina:

• Gráfico de cálculo dinâmico: O PyTorch constrói um gráfico de cálculo "em tempo real" durante a execução do código, o que facilita a depuração e permite escrever modelos mais flexíveis.
• API intuitiva e parecida com Python: a sintaxe do PyTorch é próxima do Python padrão, o que reduz a barreira de entrada e acelera o desenvolvimento.
• Amplo suporte a GPU: troca fácil entre CPU e GPU para acelerar os cálculos.
• Grande comunidade e ecossistema: muitos modelos, bibliotecas e ferramentas prontas, como torchvision, torchaudio.
• Suporte para diferenciação automática: o módulo autograd calcula automaticamente os gradientes, facilitando o treinamento de redes neurais.

Exemplo de criação de uma rede neural simples em PyTorch:

import torch
import torch.nn as nn

class SimpleNet(nn.Module):
    def __init__(self):
        super(SimpleNet, self).__init__()
        self.fc = nn.Linear(10, 1)

    def forward(self, x):
        return self.fc(x)

model = SimpleNet()
input = torch.randn(1, 10)
output = model(input)
print(output)