TensorFlow怎么创建分区变量列表

2018-09-22 17:53 更新

tf.create_partitioned_variables

create_partitioned_variables(
shape,
slicing,
initializer,
dtype=tf.float32,
trainable=True,
collections=None,
name=None,
reuse=None
)

定义在:tensorflow/python/ops/partitioned_variables.py.

根据给定的 slicing 创建分区变量的列表.

目前只能对整个变量的一个维度的实现切片 ,并且可以通过沿该维度连接返回的列表来重建完整变量.

ARGS:

  • shape:整数列表.整个变量的形状.
  • slicing:整数列表.如何对变量分区.必须有相同长度的 shape.每个值表示在相应维度中创建多少个切片.目前只有一个值可以大于 1 ;也就是说,变量只能沿着一个维度进行切片.
    为了方便起见,请求的分区数量不必均匀地分配相应的维数.如果没有,则分区的形状从分区 0 开始递增 1,直到所有的松弛都被吸收.调整规则在将来可能会改变,但是您可以使用不同的切片规格保存/恢复这些变量,这不成问题.
  • initializer:形状或变量初始化函数.如果一个函数,它将为每个切片调用一次,将切片的形状和数据类型作为参数传递.该函数必须返回与切片相同形状的张量.
  • dtype:变量的类型.如果初始值设定项为张量,则忽略.
  • trainable:如果为 True ,还将所有变量添加到图形集合 GraphKeys. TRAINABLE_VARIABLES.
  • collections:要向其中添加变量的图形集合键的列表,默认为 [GraphKeys.GLOBAL_VARIABLES].
  • name:完整变量的可选名称.默认为 "PartitionedVariable",并自动获取 uniquified.
  • reuse:是布尔值或者 None;如果为 true 且设置了名字,它将重用以前创建的变量.如果为 false,它将创建新的变量.如果为 None,它将继承父范围重用. 

返回:

返回与切片相对应的变量的列表.

注意:

  • ValueError:如果任何参数的格式都是错误的.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy