Python tf.dynamic_partition用法及代碼示例

使用來自 partitions 的索引將 data 劃分為 num_partitions 張量。

用法

tf.dynamic_partition(
    data, partitions, num_partitions, name=None
)

對於大小為 partitions.ndim 的每個索引元組 js ，切片 data[js, ...] 成為 outputs[partitions[js]] 的一部分。帶有partitions[js] = i的切片按照js的字典順序放置在outputs[i]中，outputs[i]的第一個維度是partitions中的條目數等於i。詳細地，

outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]

    outputs[i] = pack([data[js, ...] for js if partitions[js] == i])

data.shape 必須以 partitions.shape 開頭。

例如：

# Scalar partitions.
    partitions = 1
    num_partitions = 2
    data = [10, 20]
    outputs[0] = []  # Empty with shape [0, 2]
    outputs[1] = [[10, 20]]

    # Vector partitions.
    partitions = [0, 0, 1, 1, 0]
    num_partitions = 2
    data = [10, 20, 30, 40, 50]
    outputs[0] = [10, 20, 50]
    outputs[1] = [30, 40]

有關如何合並分區的示例，請參閱dynamic_stitch。