Perguntas com a marcação «tensor»

205
Como o método "view" funciona no PyTorch?

Estou confuso sobre o método view()no seguinte trecho de código. class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16*5*5, 120) self.fc2 = nn.Linear(120,...