View source on GitHub
|
Variable scope object to carry defaults to provide to get_variable.
tf.compat.v1.VariableScope(
reuse, name='', initializer=None, regularizer=None, caching_device=None,
partitioner=None, custom_getter=None, name_scope='', dtype=tf.dtypes.float32,
use_resource=None, constraint=None
)
Many of the arguments we need for get_variable in a variable store are most
easily handled with a context. This object is used for the defaults.
name: name of the current scope, used as prefix in get_variable.initializer: default initializer passed to get_variable.regularizer: default regularizer passed to get_variable.reuse: Boolean, None, or tf.compat.v1.AUTO_REUSE, setting the reuse in
get_variable. When eager execution is enabled this argument is always
forced to be False.caching_device: string, callable, or None: the caching device passed to
get_variable.partitioner: callable or None: the partitioner passed to get_variable.custom_getter: default custom getter passed to get_variable.name_scope: The name passed to tf.name_scope.dtype: default type passed to get_variable (defaults to DT_FLOAT).use_resource: if False, create a normal Variable; if True create an
experimental ResourceVariable with well-defined semantics. Defaults to
False (will later change to True). When eager execution is enabled this
argument is always forced to be True.constraint: An optional projection function to be applied to the variable
after being updated by an Optimizer (e.g. used to implement norm
constraints or value constraints for layer weights). The function must
take as input the unprojected Tensor representing the value of the
variable and return the Tensor for the projected value (which must have
the same shape). Constraints are not safe to use when doing asynchronous
distributed training.* original_name_scopeget_collectionget_collection(
name
)
Get this scope's variables.
get_variableget_variable(
var_store, name, shape=None, dtype=None, initializer=None, regularizer=None,
reuse=None, trainable=None, collections=None, caching_device=None,
partitioner=None, validate_shape=True, use_resource=None, custom_getter=None,
constraint=None, synchronization=tf.VariableSynchronization.AUTO,
aggregation=tf.compat.v1.VariableAggregation.NONE
)
Gets an existing variable with this name or create a new one.
global_variablesglobal_variables()
Get this scope's global variables.
local_variableslocal_variables()
Get this scope's local variables.
reuse_variablesreuse_variables()
Reuse variables in this scope.
set_caching_deviceset_caching_device(
caching_device
)
Set caching_device for this scope.
set_custom_getterset_custom_getter(
custom_getter
)
Set custom getter for this scope.
set_dtypeset_dtype(
dtype
)
Set data type for this scope.
set_initializerset_initializer(
initializer
)
Set initializer for this scope.
set_partitionerset_partitioner(
partitioner
)
Set partitioner for this scope.
set_regularizerset_regularizer(
regularizer
)
Set regularizer for this scope.
set_use_resourceset_use_resource(
use_resource
)
Sets whether to use ResourceVariables for this scope.
trainable_variablestrainable_variables()
Get this scope's trainable variables.