tf_agents.networks.network.get_state_spec
Stay organized with collections
Save and categorize content based on your preferences.
Extracts the state spec from a layer.
tf_agents.networks.network.get_state_spec(
layer: tf.keras.layers.Layer
) -> tf_agents.typing.types.NestedTensorSpec
Args |
|---|
layer
|
The layer to extract from; can be a Network.
|
Raises |
|---|
TypeError
|
If layer is a subclass of tf.keras.layers.RNN (it must
be wrapped by an RNNWrapper object).
|
ValueError
|
If layer is a Keras layer and create_variables has
not been called on it.
|
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."],[],[],null,["# tf_agents.networks.network.get_state_spec\n\n\u003cbr /\u003e\n\n|--------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/networks/network.py#L715-L752) |\n\nExtracts the state spec from a layer. \n\n tf_agents.networks.network.get_state_spec(\n layer: tf.keras.layers.Layer\n ) -\u003e ../../../tf_agents/typing/types/NestedTensorSpec\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|---------|------------------------------------------------|\n| `layer` | The layer to extract from; can be a `Network`. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| The state spec. ||\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Raises ------ ||\n|--------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `TypeError` | If `layer` is a subclass of [`tf.keras.layers.RNN`](https://www.tensorflow.org/api_docs/python/tf/keras/layers/RNN) (it must be wrapped by an `RNNWrapper` object). |\n| `ValueError` | If `layer` is a Keras layer and `create_variables` has not been called on it. |\n\n\u003cbr /\u003e"]]
View source on GitHub