variable
tflearn.variables.variable (args, *kwargs)
实例化一个新变量。
参数
- name:
str
。变量的名称。 - shape:
int
列表。变量的形状(可选)。 - dtype:
type
。变量的数据类型。 - initializer:
str
或Tensor
。变量的初始化方法。(请参阅 tflearn.initializations 获取参考)。 - regularizer:
str
或Tensor
。变量的正则化器。(请参阅 tflearn.losses 获取参考)。 - trainable:
bool
。如果为 True,则训练此变量的权重。 - collections:
str
。要将新变量添加到的集合(可选)。 - caching_device:
str
。可选的设备字符串或函数,用于描述应在何处缓存变量以供读取。默认为变量的设备。 - validate_shape:
bool
。还原时是否验证形状。 - device:
str
。用于存储变量的可选设备 ID。 - restore:
bool
。加载预训练模型时是否还原此变量(仅与 tflearn 预构建的训练函数兼容)。
返回值
一个变量。
get_all_variables
tflearn.variables.get_all_variables ()
获取所有图形变量。
返回值
变量列表。
get_all_variables
tflearn.variables.get_all_trainable_variable ()
获取所有图形可训练变量。
返回值
变量列表。
get_layer_variables_by_name
tflearn.variables.get_layer_variables_by_name (name)
根据名称检索层的变量。
参数
- name:
str
。层的名称。
返回值
变量列表。
get_value
tflearn.variables.get_value (var, session=None)
获取变量的值。如果没有提供会话,则使用默认会话。
参数
- var:
Variable
。要获取值的变量。 - session:
Session
。用于运行操作的会话。默认值:默认会话。
返回值
变量的值。
set_value
tflearn.variables.set_value (var, value, session=None)
设置变量的值。如果没有提供会话,则使用默认会话。
参数
- var:
Variable
。要分配值的变量。 - value: 要分配的值。必须与变量数据类型兼容。
- session:
Session
。用于执行分配的会话。默认值:默认会话。