如何獲取圖層的形狀/尺寸？

Question

當我已經定義了這樣的模式：

def create_basic_model_terse(input, out_dims): 

    with default_options(activation=relu): 
     model = Sequential([ 
      LayerStack(3, lambda i: [ 
       Convolution((5,5), [32,32,64][i], init=glorot_uniform(), pad=True), 
       MaxPooling((3,3), strides=(2,2)) 
      ]), 
      Dense(64, init=glorot_uniform()), 
      Dense(out_dims, init=glorot_uniform(), activation=None) 
     ]) 

    return model(input) 

我怎樣才能獲得某種關於網絡狀輸出形狀/大小在每一層的信息？

Answer 1

你可以看看CNTK 202教程。還有其他一些教程，如CNTK 105也展示瞭如何獲得不同的模型屬性。

For a model 
 
def create_model(): 
 
with default_options(initial_state=0.1): 
 
    return Sequential([ 
 
     Embedding(emb_dim), 
 
     Recurrence(LSTM(hidden_dim), go_backwards=False), 
 
     Dense(num_labels) 
 
    ]) 
 

 

 

 
model = create_model() 
 
print(len(model.layers)) 
 
print(model.layers[0].E.shape) 
 
print(model.layers[2].b.value)