View source on GitHub
|
Utilities related to TensorSpec class.
Functions
add_outer_dim(...): Adds an outer dimension to the shape of input specs.
add_outer_dims_nest(...): Adds outer dimensions to the shape of input specs.
from_packed_proto(...): Decodes a packed Any proto containing the structured value for the spec.
from_pbtxt_file(...): Loads a spec encoded as a struct_pb2.StructuredValue from a pbtxt file.
from_proto(...): Decodes a struct_pb2.StructuredValue proto into a nested spec.
from_spec(...): Maps the given spec into corresponding TensorSpecs keeping bounds.
remove_outer_dims_nest(...): Removes the specified number of outer dimensions from the input spec nest.
sample_bounded_spec(...): Samples uniformily the given bounded spec.
sample_spec_nest(...): Samples the given nest of specs.
to_array_spec(...): Converts TensorSpec into ArraySpec.
to_nest_array_spec(...): (Deprecated) Alias for to_array_spec.
to_nest_placeholder(...): Converts a nest of TensorSpecs to a nest of matching placeholders.
to_pbtxt_file(...): Saves a spec encoded as a struct_pb2.StructuredValue in a pbtxt file.
to_placeholder(...): Creates a placeholder from TensorSpec.
to_placeholder_with_default(...): Creates a placeholder from TensorSpec.
to_proto(...): Encodes a nested spec into a struct_pb2.StructuredValue proto.
with_dtype(...): Updates dtypes of all specs in the input spec.
zero_spec_nest(...): Create zero tensors for a given spec.
Other Members | |
|---|---|
| absolute_import |
Instance of __future__._Feature
|
| division |
Instance of __future__._Feature
|
| print_function |
Instance of __future__._Feature
|