View source on GitHub |
Checks that output_spec
is a nest of "value" type values.
tf_agents.utils.nest_utils.assert_value_spec(
output_spec: tf_agents.typing.types.NestedTensorSpec
,
network_name: str
)
"value" type values correspond to floating point tensors with spec shape
()
or (1,)
.
Args | |
---|---|
output_spec
|
The output spec returned by network.create_variables .
|
network_name
|
The string name of the network for error messages. |
Raises | |
---|---|
ValueError
|
If output_spec is not a nest of value-type tensors.
|