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.

參見指南:變量>切分變量分區(qū)

分區(qū)為每個切片分配最小尺寸.
返回一個分區(qū),該值對給定形狀的變量和 dtype 進行分區(qū),這樣每個分區(qū)都有最小的 min_slice_size 切片的變量.此類分區(qū)的最大數目 (上限) 由 max_partitions 提供.

參數:

  • max_partitions:分區(qū)數的上限,默認為 1.
  • axis:沿其對變量進行分區(qū)的軸,默認為 0.
  • min_slice_size:每個分區(qū)的可變切片的最小尺寸,默認為 256K.
  • bytes_per_string_element:如果變量是 string 類型的,這將提供一個估計變量中每個標量的大小.

返回值:

可用作variable_scopeget_variable, 和 get_partitioned_variable_list分區(qū)參數的分區(qū)函數.


以上內容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號