o ?e@sdZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZd Zd Ze jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZd dZdddZddZddZdS)avModule implementing RNN Cells. This module provides a number of basic commonly used RNN cells, such as LSTM (Long Short Term Memory) or GRU (Gated Recurrent Unit), and a number of operators that allow adding dropouts, projections, or embeddings for inputs. Constructing multi-layer cells is supported by the class `MultiRNNCell`, or by calling the `rnn` ops several times. )context) constant_op)dtypes)tensor) tensor_shape) tensor_util) rnn_cell_impl) array_ops)nestZbiasZkernelcsfdd}t||S)zCCreate tensors of zeros based on state_size, batch_size, and dtype.cs<t|}tj|d}tst|dd}|||S)z7Combine s with batch_size to get a proper tensor shape.dtypeT)static)_concatr ZzerosrZexecuting_eagerly set_shape)scsizeZc_static batch_sizer d/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/rnn_cell_impl.pyget_state_shape5s  z,_zero_state_tensors..get_state_shape)r Z map_structure) state_sizerr rrrr_zero_state_tensors2s rFcCst|tjr'|}t|}|jjdkrt|d}qG|jjdkr&t d|n t |}|jdur5| nd}| rEtj| tjdnd}t|tjrn|}t|}|jjdkrat|d}q|jjdkrmt d|n t |}|jdur|| nd}| rtj| tjdnd}|rt ||}|jdur| }|Sd}|S|dus|durt d||ft||fd}|S)aConcat that enables int, Tensor, or TensorShape values. This function takes a size specification, which can be an integer, a TensorShape, or a Tensor, and converts it into a concatenated Tensor (if static = False) or a list of integers (if static = True). Args: prefix: The prefix; usually the batch size (and/or time step size). (TensorShape, int, or Tensor.) suffix: TensorShape, int, or Tensor. static: If `True`, return a python list with possibly unknown dimensions. Otherwise return a `Tensor`. Returns: shape: the concatenation of prefix and suffix. Raises: ValueError: if `suffix` is not a scalar or vector (or TensorShape). ValueError: if prefix or suffix was `None` and asked for dynamic Tensors out. rzCprefix tensor must be either a scalar or vector, but saw tensor: %sNr zCsuffix tensor must be either a scalar or vector, but saw tensor: %sz.Provided a prefix or suffix of None: %s and %s) isinstancerZTensorrZconstant_valueshapeZndimsr Z expand_dims ValueErrorrZ TensorShapeas_listZis_fully_definedrZconstantrZint32Z concatenateconcat)prefixsuffixr pZp_staticrZs_staticrrrrrAs`           rcCs&zt||WdStyYdSw)NFT)getattrAttributeError)objZ attr_namerrr_hasattrs   r&cCslt|dt|dt|dpt|dt|g}gd}t|s4ddt||D}td||d |d S) aRaises a TypeError if cell is not like an RNNCell. NOTE: Do not rely on the error message (in particular in tests) which can be subject to change to increase readability. Use ASSERT_LIKE_RNNCELL_ERROR_REGEXP. Args: cell_name: A string to give a meaningful error referencing to the name of the functionargument. cell: The object which should behave like an RNNCell. Raises: TypeError: A human-friendly exception. Z output_sizerZget_initial_stateZ zero_state)z!'output_size' property is missingz 'state_size' property is missingz=either 'zero_state' or 'get_initial_state' method is requiredzis not callablecSsg|]\}}|s|qSrr).0errorZcondrrr sz'assert_like_rnncell..z-The argument {!r} ({}) is not an RNNCell: {}.z, N)r&callableallzip TypeErrorformatjoin)Z cell_namecellZ conditionserrorsrrrassert_like_rnncells r2N)F) __doc__Ztensorflow.python.eagerrZtensorflow.python.frameworkrrrrrZ)tensorflow.python.keras.layers.legacy_rnnrZtensorflow.python.opsr Ztensorflow.python.utilr Z_BIAS_VARIABLE_NAMEZ_WEIGHTS_VARIABLE_NAMEZ BasicLSTMCellZ BasicRNNCellZ DeviceWrapperZDropoutWrapperZGRUCellZ LayerRNNCellZLSTMCellZLSTMStateTupleZ MultiRNNCellZResidualWrapperZRNNCellrrr&r2rrrrs6          G