2017-11-24 1 views
1

Mxnet 프레임 워크를 이해하고 있습니다. 반면 다음 나는 고밀도 레이어 생성되는 코드의 다음 부분을 참조이 Linear Regression article :Mxnet - 밀집 층의 무게 모양 이해

net = gluon.nn.Dense(1, in_units=2) # input dimension= 2, output dimension = 1 

을하지만 왜 print(net.weight)이 모양이 (2, 1)이어야한다 Parameter dense4_weight (shape=(1, 2), dtype=None)

같은 모양을 주는가를? 내 이해에 따르면

:

input = Shape(n, 2) where n is number of samples 
output = Shape(n, 1) 

그렇게 가중 매트릭스 형상 (2, 1)의 행렬 곱셈, 그렇지되어야 하는가?

무엇이 여기에 있습니까?

답변

0

이것은 단지 내부 구현 때문입니다. 가중치 행렬이 곱하기 전에 전치됨에 따라 생각할 수 있습니다.