variable

tflearn.variables.variable (args, *kwargs)

实例化一个新变量。

参数

  • name: str。变量的名称。
  • shape: int 列表。变量的形状(可选)。
  • dtype: type。变量的数据类型。
  • initializer: strTensor。变量的初始化方法。(请参阅 tflearn.initializations 获取参考)。
  • regularizer: strTensor。变量的正则化器。(请参阅 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。用于执行分配的会话。默认值:默认会话。