tf_agents.networks.network.create_variables
Stay organized with collections
Save and categorize content based on your preferences.
Create variables in module
given input_spec
; return output_spec
.
tf_agents.networks.network.create_variables(
module: typing.Union[Network, tf.keras.layers.Layer],
input_spec: typing.Optional[types.NestedTensorSpec] = None,
**kwargs
) -> tf_agents.typing.types.NestedTensorSpec
Here module
can be a tf_agents.networks.Network
or Keras
layer.
Args |
module
|
The instance we would like to create layers on.
|
input_spec
|
The input spec (excluding batch dimensions).
|
**kwargs
|
Extra arguments to module.__call__ , e.g. training=True .
|
Raises |
ValueError
|
If module is a generic Keras layer but input_spec is None .
|
TypeError
|
If module is a tf.keras.layers.{RNN,LSTM,GRU,...} . These
must be wrapped in tf_agents.keras_layers.RNNWrapper .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-04-26 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-04-26 UTC."],[],[]]