tf_agents.typing.types.NestedSpecTensorOrArray

This symbol is a type alias.

Source:

NestedSpecTensorOrArray = Union[
    tensorflow.python.framework.type_spec.TypeSpec,
    tensorflow.python.framework.tensor.TensorSpec,
    tensorflow.python.ops.ragged.ragged_tensor.RaggedTensorSpec,
    tensorflow.python.framework.sparse_tensor.SparseTensorSpec,
    tf_agents.types.DistributionSpecV2,
    Iterable[ForwardRef('NestedTensorSpec')],
    Mapping[str, ForwardRef('NestedTensorSpec')],
    tf_agents.types.ArraySpec,
    Iterable[ForwardRef('NestedArraySpec')],
    Mapping[str, ForwardRef('NestedArraySpec')],
    tensorflow.python.framework.tensor.Tensor,
    tensorflow.python.framework.sparse_tensor.SparseTensor,
    tensorflow.python.ops.ragged.ragged_tensor.RaggedTensor,
    tensorflow.python.framework.ops.EagerTensor,
    Iterable[ForwardRef('NestedTensor')],
    Mapping[str, ForwardRef('NestedTensor')],
    numpy.ndarray,
    int,
    float,
    str,
    bool,
    Iterable[ForwardRef('NestedArray')],
    Mapping[str, ForwardRef('NestedArray')]
]