Un'operazione che raggruppa insieme un elenco di input partizionati. Supporta lo sharding ND.
Classi nidificate
classe | TPUPartitionedInputV2.Options | Attributi facoltativi per TPUPartitionedInputV2 |
Metodi pubblici
Uscita <T> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
<T> statico TPUPartitionedInputV2 <T> | create (ambito di applicazione , ingressi Iterable< operando <T>>, list<lungo> partizioneDims, opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione TPUPartitionedInputV2. |
TPUPartitionedInputV2.Options statico | isPacked (valore booleano isPacked) |
Uscita <T> | produzione () Una maniglia che rappresenta la forma completa di tensori partizionati. |
Metodi ereditati
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce la maniglia simbolica di un tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static TPUPartitionedInputV2 <T> create ( scope scope, Iterable< Operand <T>> input, List<Long> partitionDims, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione TPUPartitionedInputV2.
Parametri
scopo | ambito attuale |
---|---|
input | Un elenco di input partizionati che devono avere la stessa forma. |
partizioneDims | Un elenco di numeri interi che descrivono come è partizionata ciascuna dimensione. Il vuoto indica che gli input vengono replicati. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di TPUPartitionedInputV2
pubblico statico TPUPartitionedInputV2.Options isPacked (Booleano isPacked)
Parametri
è pieno | Indica se l'input è una risorsa compressa. |
---|
Uscita pubblica <T> uscita ()
Una maniglia che rappresenta la forma completa di tensori partizionati.