dnn.py 文件源码

def get_weights(self, weight_tensor):
        """ Get Weights.

        Get a variable weights.

        Examples:

dnn = DNNTrainer(...)
        w = dnn.get_weights(denselayer.W) # get a dense layer weights
        w = dnn.get_weights(convlayer.b) # get a conv layer biases
        ```

    Arguments:
        weight_tensor: `Tensor`. A Variable.

    Returns:
        `np.array`. The provided variable weights.
    """
    return weight_tensor.eval(self.trainer.session)

```