Crea un TensorArray per l'archiviazione di più gradienti di valori nell'handle specificato.
Simile a TensorArrayGradV3. Tuttavia crea un accumulatore con una forma espansa rispetto all'input TensorArray di cui viene calcolato il gradiente. Ciò consente di calcolare più gradienti per lo stesso TensorArray utilizzando lo stesso accumulatore.
Metodi pubblici
static TensorArrayGradWithShape | |
Uscita <Float> | flowOut () |
Uscita <?> | gradHandle () |
Metodi ereditati
Metodi pubblici
public static TensorArrayGradWithShape create ( ambito ambito, operatore <?> handle, operando <Float> flowIn, operando <Integer> shapeToPrepend, stringa sorgente)
Metodo Factory per creare una classe che racchiude una nuova operazione TensorArrayGradWithShape.
Parametri
scopo | ambito attuale |
---|---|
maniglia | L'handle del TensorArray in avanti. |
flowIn | Uno scalare float che impone il corretto concatenamento delle operazioni. |
shapeToPrepend | Un vettore int32 che rappresenta una forma. Gli elementi nell'accumulatore gradiente avranno una forma che è questo valore shape_to_prepend concatenato con la forma degli elementi nel TensorArray corrispondente all'handle di input. |
fonte | La stringa di origine del gradiente, utilizzata per decidere quale gradiente TensorArray restituire. |
ritorna
- una nuova istanza di TensorArrayGradWithShape