2Vh@ddlZddlmZddlmZddlmZddlmZddl m Z edGdd e Z d Z d Z d Zd Zy)N)tree) keras_export KerasTensor)backend) Operationzkeras.FunctioncteZdZdZd fd ZedZedZedZdZ dZ dZ d d Z d Z xZS) FunctionaClass that encapsulates a computation graph of Keras operations. You can use a `Function` to capture the computation graph linking some input tensors to some output tensors, and reapply the same computation on new inputs. A `Function` is similar to a Functional Model, with the difference that it is stateless (it does not track state variables) and does not implement the `Layer` API. Example: ```python input_1 = keras.KerasTensor(shape=(None, 2, 3)) input_2 = keras.KerasTensor(shape=(None, 2, 3)) x = input_1 + input_2 output = keras.ops.sigmoid(x) fn = keras.Function(inputs=[input_1, input_2], outputs=output) input_1_val = np.random.random((4, 2, 3)) input_2_val = np.random.random((4, 2, 3)) output_val = fn([input_1_val, input_2_val]) ``` Args: inputs: `KerasTensor` instance or nested structured of `KerasTensor` instances. outputs: `KerasTensor` instance or nested structured of `KerasTensor` instances. They should be computable given only the values of `inputs`. name: String. The name of the function. ct ||tdk(rt|dd}d|_t j d||_t j d||_t j||_ t j||_ |jstd|d ||jstd |d |tdk(r|_t|j|j\}}}}||_||_||_||_|jD]D} | j$j&s| j$j&j(r;td y) N)name tensorflow_self_setattr_trackingTFc|SNxs F/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/ops/function.pyz#Function.__init__..9s1c|Srrrs rrz#Function.__init__..:sArz4`inputs` argument cannot be empty. Received: inputs=z outputs=z5`outputs` argument cannot be empty. Received: inputs=z#`inputs` not connected to `outputs`)super__init__rgetattrrr map_structure_inputs_struct_outputs_structflatten_inputs_outputs ValueError map_graph_nodes_nodes_by_depth _operations_operations_by_depth_keras_history operation_outbound_nodes) selfinputsoutputsr rnodesnodes_by_depth operationsoperations_by_depthinput __class__s rrzFunction.__init__-s~ d# 9 $ &-.& "+0D '"00fE#11+wG||F+  W- || "")%  }} "")%  9 $*@D 'CL LL$--D @ ,? -%$7!\\ HE$$..,,66FF !FGG  Hrc |jddSr)r%r*s rr/zFunction.operations[s""rc|jS)z1Flat list of the symbolic inputs of the Function.)rr4s rr+zFunction.inputs_s||rc|jS)z2Flat list of the symbolic outputs of the Function.)r r4s rr,zFunction.outputsds}}rc2|j|d}ttj||jD]"\}}|j |j k7s d}n|r!tj d|jS|j|dS)NTFcDt|j|jS)N)shapedtype)rr9r:rs rrz.Function.compute_output_spec..ts+AGG177Crc|jSr)compute_output_specops rrz.Function.compute_output_spec..|s B,B,Br operation_fn) _assert_input_compatibilityziprrrr9rr_run_through_graph)r*r+shortcutrx_refs rr<zFunction.compute_output_specis ((0DLL0$,,? HAuww%++%   %%C$$ && !B'  rc&tjd|}ttj||jD]'\}}|j |_|j|_)|j|}tjd|S)Nct|S)Nr9rrs rrz/Function.compute_output_shape..s k*rc|jSrrHrs rrz/Function.compute_output_shape..s AGGr) rmap_shape_structurerBrrr:_dtypesparse_sparser<r)r* input_shapeinput_shape_structrrE output_specs rcompute_output_shapezFunction.compute_output_shapes"55 *K  DLL);.srrr?)rArC)r*r+s rcallz Function.calls& ((0&&vM&JJrc:tj|}it|j|D]\}}|t |<|j }t |j}|jd|D]}||} | D]} | jr | jrtfd| jDr;| jj\} } || j} | || g| i| }n| | i| }t| jtj|D]\}}|t |<g}|jD]}|j!t |!tj"|j$|S)zExecute the graph. At each node we compute outputs via `operation_fn(node.operation)(*args, **kwargs)`. Treversec38K|]}t|vywr)id).0r tensor_dicts r z.Function._run_through_graph..sLAr!uK/Ls)rrrBr+rYr$listkeyssortr(is_inputany input_tensors argumentsfill_inr,appendpack_sequence_asr)r*r+r@call_fnryr. depth_keysdepthr-nodeargskwargsr>r,output_tensorsr[s @rrCzFunction._run_through_graphs f%  V, #DAq!"K1  #--.--/0 % +E"5)E +~~L9K9KLL#~~55kB f!$..1&%b:4:6:G $1&1G  dll7.CD+DAq)*K1&+ + +( 6A  ! !+be"4 5 6$$T%9%9>JJrc  tj||jt tj ||j D]\}}t|jt|jk7rIt|jjd|jd|jd|jdt |j|jD]Y\}}| | ||k7st|jjd|jd|jd|jdy#t$rtd|jd|wxYw)NzOFunction was called with an invalid input structure. Expected input structure: z Received input structure: z, was passed incompatible inputs. For input 'z', expected shape z+, but received instead a tensor with shape .) rassert_same_structurerr!rBrrlenr9r2__name__r )r*r+rrEdimref_dims rrAz$Function._assert_input_compatibilitysr   & &vt/B/B CDLL0$,,? HAu177|s5;;// ~~../077 !$AGGU[[ 9  W&3?g~(#~~6678??DzzlK..3kk];;;<77)1F    --1-@-@,AB--3H6  s E%E)r)rs __module__ __qualname____doc__rpropertyr/r+r,r<rQrTrCrA __classcell__)r2s@rr r siB,H\## , BK )KVrr cHtt|dzt|zS)Nz_ib-)strrY)r> node_indexs r make_node_keyr~s r"v; #j/ 11rc &t||\}|Dchc];}t|j|jjj |=}}i}i}t |D]}|j |d}|j|jd}t||}|||j<|||<|jD]&} |j| d}t|dz||| <(|D]P} | jd} | s| |vsd|| <d| <d|| jd<|jt| dRtjt} |jD]\}}| |j!|tjt} |jD]\}}| |j!|t| j#}|j%dg}|D]-}| |}|j%fd|j'|/t| j#}|j%dt)}|D]}|j|g}|D]}| |D]}t+j,|j.D]K}||vr |j}t1d|d |d ||j!|jj2Mt+j,|j4D]}|j||Dcgc]}|j2}}|D]7}|j7|dk7st1d |d |j7|d || || fScc}wcc}w)a Validates a graph's topology and gather its operations and nodes. Args: inputs: List of input tensors. outputs: List of outputs tensors. Returns: A tuple `(nodes, nodes_by_depth, operations, operations_by_depth)`. - nodes: set of Node instances - nodes_by_depth: dict mapping ints (depth) to lists of node instances. - operations: list of Operation instances. - operations_by_depth: dict mapping ints (depth) to lists of Operation instances. rTrVc|Srr)roperation_indicess rrzmap_graph..+s 0A!0Dr)keyz2Graph disconnected: cannot find parent for tensor z at operation 'zB'. The following previous operations were accessed without issue: z The name "z " is used z: times in the model. All operation names should be unique.) _build_mapr~r(_inbound_nodesindexreversed setdefaultgetmax parent_nodesr'add collections defaultdictr]itemsrer^r_extendsetrrrbr!r r,count)r+r,nodes_in_decreasing_depthrk network_nodes nodes_depthsoperations_depthsrjprevious_depthnode_depinput_tinput_operationr.r0r(rir/operations_for_depthcomputable_tensorsroperations_with_complete_input all_namesr rs @rr"r"s "4>fg3N00.  dnndnn&C&C&I&I$&OPM L23D''a0+..t~~qA E>*,1$..)" T )) DH)--h:N%(N%CL " D#D0A!003 6GG12 o .13 o .>?L77: ;   mOQ? @ A!,,T2N#))++ eu$$T*+&11$7-3355 5E")))45)..01JOODO!J0259 !!&D!E./ 0n))+,JOODO!  "q!"&("*"5) *D\\$"4"45 K.. $I$""#OI;?**H)IK /55dnn6I6IJ K\\$,,/ *"&&q) * **$2<?   ))'*+<'=)$$ ==V4<<+??(( F !)!   tT"$$T*r)r keras.srcrkeras.src.api_exportrkeras.src.backendrkeras.src.backend.configrkeras.src.ops.operationrr r~r"rrrrrrsT-),-KyK K\2vJr#8L1+r