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_scope
, get_variable
, 和 get_partitioned_variable_list
分区参数的分区函数.
以上内容是否对您有帮助:
更多建议: