MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / __init__

Method __init__

modelzoo/dien/train.py:51–63  ·  view source on GitHub ↗
(self,
                 num_units,
                 activation=None,
                 reuse=None,
                 kernel_initializer=None,
                 bias_initializer=None)

Source from the content-addressed store, hash-verified

49
50class VecAttGRUCell(tf.nn.rnn_cell.RNNCell):
51 def __init__(self,
52 num_units,
53 activation=None,
54 reuse=None,
55 kernel_initializer=None,
56 bias_initializer=None):
57 super(VecAttGRUCell, self).__init__(_reuse=reuse)
58 self._num_units = num_units
59 self._activation = activation or tf.math.tanh
60 self._kernel_initializer = kernel_initializer
61 self._bias_initializer = bias_initializer
62 self._gate_linear = None
63 self._candidate_linear = None
64
65 @property
66 def state_size(self):

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected