阅读(9.8k) 书签 (0)

TensorFlow变量:tf.min_max_variable_partitioner

2018-11-03 11:42 更新
tf.min_max_variable_partitioner 函数
min_max_variable_partitioner(
    max_partitions=1,
    axis=0,
    min_slice_size=256 << 10,
    bytes_per_string_element=16
)

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

参见指南:变量>切分变量分区

分区为每个切片分配最小尺寸.
返回一个分区,该值对给定形状的变量和 dtype 进行分区,这样每个分区都有最小的 min_slice_size 切片的变量.此类分区的最大数目 (上限) 由 max_partitions 提供.

参数:

  • max_partitions:分区数的上限,默认为 1.
  • axis:沿其对变量进行分区的轴,默认为 0.
  • min_slice_size:每个分区的可变切片的最小尺寸,默认为 256K.
  • bytes_per_string_element:如果变量是 string 类型的,这将提供一个估计变量中每个标量的大小.

返回值:

可用作variable_scopeget_variable, 和 get_partitioned_variable_list分区参数的分区函数.