Particiona `data` em tensores `num_partitions` usando índices de `partitions`.
Para cada tupla de índice `js` de tamanho `partitions.ndim`, a fatia `data[js, ...]` torna-se parte de `outputs[partitions[js]]`. As fatias com `partitions[js] = i` são colocadas em `outputs[i]` na ordem lexicográfica de `js`, e a primeira dimensão de `outputs[i]` é o número de entradas em `partitions` igual a `eu`. Em detalhes,
outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]
outputs[i] = pack([data[js, ...] for js if partitions[js] == i])
Por exemplo:
# 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]
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático <T estende TType > DynamicPartition <T> | criar (escopo do escopo , dados do operando <T>, partições do operando <TInt32> , NumPartitions longas) Método de fábrica para criar uma classe que envolve uma nova operação DynamicPartition. |
Iterador< Operando <T>> | iterador () |
Lista< Saída <T>> | saídas () |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
public static DynamicPartition <T> create (escopo do escopo , dados do operando <T>, partições do operando <TInt32> , numPartitions longas)
Método de fábrica para criar uma classe que envolve uma nova operação DynamicPartition.
Parâmetros
escopo | escopo atual |
---|---|
partições | Qualquer formato. Índices no intervalo `[0, num_partitions)`. |
numPartições | O número de partições para saída. |
Devoluções
- uma nova instância de DynamicPartition