![]() |
Belle II Software development
|
Public Member Functions | |
__init__ (self, name, tf_activation_str, dim_input, dim_output, p_bias, p_w, random_seed=None) | |
__call__ (self, x) | |
variable_to_summary (self, var, step, writer) | |
all_to_summary (self, step, writer) | |
Public Attributes | |
tf_activation = tf_activation_dict[tf_activation_str] | |
activation function | |
list | shape = [dim_input, dim_output] |
layer shape | |
w = self._init_weight(self.shape, p_w, random_seed) | |
init parameters for uniform distribution | |
b = self._init_bias(self.shape[1], p_bias) | |
init parameters for bias | |
input = None | |
input | |
output = None | |
output | |
Protected Member Functions | |
_init_bias (self, width, init_val, name=None) | |
_init_weight (self, shape, stddev, operation_seed, name=None) | |
definition of a layer obj
Definition at line 21 of file tensorflow_dnn_model.py.
__init__ | ( | self, | |
name, | |||
tf_activation_str, | |||
dim_input, | |||
dim_output, | |||
p_bias, | |||
p_w, | |||
random_seed = None ) |
:param name: name of the layer. :param tf_activation: string, name of an available tensorflow activations function :param dim_input: dimension of the input :param dim_output: dimension of the output :param p_bias: initial bias :param p_w: stddev of uniform distribution to initialize :param random_seed: random seed used in initialising the weights :return: None
Definition at line 26 of file tensorflow_dnn_model.py.
__call__ | ( | self, | |
x ) |
evaluate the layer
Definition at line 88 of file tensorflow_dnn_model.py.
|
protected |
define bias variables
Definition at line 69 of file tensorflow_dnn_model.py.
|
protected |
define weight variables
Definition at line 78 of file tensorflow_dnn_model.py.
all_to_summary | ( | self, | |
step, | |||
writer ) |
Passes all layer variables to the tf.summary writer.
Definition at line 109 of file tensorflow_dnn_model.py.
variable_to_summary | ( | self, | |
var, | |||
step, | |||
writer ) |
Passes information about each variable to the summary writer.
Definition at line 94 of file tensorflow_dnn_model.py.
b = self._init_bias(self.shape[1], p_bias) |
init parameters for bias
Definition at line 61 of file tensorflow_dnn_model.py.
input = None |
input
Definition at line 64 of file tensorflow_dnn_model.py.
output = None |
output
Definition at line 67 of file tensorflow_dnn_model.py.
list shape = [dim_input, dim_output] |
layer shape
Definition at line 55 of file tensorflow_dnn_model.py.
tf_activation = tf_activation_dict[tf_activation_str] |
activation function
Definition at line 52 of file tensorflow_dnn_model.py.
w = self._init_weight(self.shape, p_w, random_seed) |
init parameters for uniform distribution
Definition at line 58 of file tensorflow_dnn_model.py.