View source on GitHub |
Make list of unstacked nested tensors.
tf_agents.utils.nest_utils.unstack_nested_tensors(
tensors, specs
)
Args | |
---|---|
tensors
|
Nested tensors whose first dimension is to be unstacked. |
specs
|
Tensor specs for tensors. |
Returns | |
---|---|
A list of the unstacked nested tensors. |
Raises | |
---|---|
ValueError
|
if the tensors and specs have incompatible dimensions or shapes. |