인프런 커뮤니티 질문&답변

주피터 노트북 "06_1_DNN_PYTORCH" 자료에서 등록된 parameter 의 차원을 출력해보면 최초 출력되는 파라미터가 최초 등록한 nn.Linear(input_dim, 10) // (4, 10) 이면 (4, 10) 이 출력되어야 할 것 같은데 왜 (10, 4) 가 출력될까요?

x = torch.ones(4) # input tensor

y = torch.zeros(3) # expected output

input_dim = x.size(0)

output_dim = y.size(0)

model = nn.Sequential (

nn.Linear(input_dim, 10),

nn.LeakyReLU(0.1),

nn.Linear(10, 10),

nn.LeakyReLU(0.1),

nn.Linear(10, 10),

nn.LeakyReLU(0.1),

nn.Linear(10, output_dim)

)

loss_function = nn.MSELoss()

learning_rate = 0.01

nb_epochs = 1000

optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)

for epoch in range(nb_epochs + 1):

y_pred = model(x)

loss = loss_function(y_pred, y)

optimizer.zero_grad()

loss.backward()

optimizer.step()

print(loss)

for param in model.parameters():

print (param)

print(param.shape)