How do I get the weights of a layer in Keras?

前端 未结 4 378
误落风尘
误落风尘 2020-12-08 02:38

I am using Windows 10, Python 3.5, and tensorflow 1.1.0. I have the following script:

import tensorflow as tf
import tensorflow.contrib.keras.api.keras.backe         


        
4条回答
  •  暖寄归人
    2020-12-08 03:33

    If you write:

    dense1 = Dense(10, activation='relu')(input_x)

    Then dense1 is not a layer, it's the output of a layer. The layer is Dense(10, activation='relu')

    So it seems you meant:

    dense1 = Dense(10, activation='relu')
    y = dense1(input_x)
    

    Here is a full snippet:

    import tensorflow as tf
    from tensorflow.contrib.keras import layers
    
    input_x = tf.placeholder(tf.float32, [None, 10], name='input_x')    
    dense1 = layers.Dense(10, activation='relu')
    y = dense1(input_x)
    
    weights = dense1.get_weights()
    

提交回复
热议问题