Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.60 KB | None | 0 0
  1. rnnTimeStep[params_Association, input_, state_] :=
  2. Tanh[params["StateWeights"].state + params["InputWeights"].input +
  3. params["Biases"]];
  4. rnnLayerForward[params_Association, input_, state_] :=
  5. Rest@FoldList[rnnTimeStep[params, #2, #1] &, state, input]
  6.  
  7. sequenceLength = 3;
  8. featureSize = 2;
  9. rnnStateSize = 1;
  10. inData = RandomReal[1, {sequenceLength, featureSize}];
  11. state = ConstantArray[0, rnnStateSize];
  12.  
  13. net = NetInitialize@
  14. BasicRecurrentLayer[rnnStateSize, "Input" -> Dimensions@inData]
  15.  
  16. param = Normal@NetExtract[net, "Arrays"]
  17.  
  18. net[inData]
  19.  
  20. rnnLayerForward[param, inData, state]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement