TensorArrayGradWithShape

public final class TensorArrayGradWithShape

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
create ( ambito ambito, operatore <?> handle, operando <Float> flowIn, operando <Integer> shapeToPrepend, String source)
Metodo Factory per creare una classe che racchiude una nuova operazione TensorArrayGradWithShape.
Uscita <Float>
Uscita <?>

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

output pubblico <Float> flowOut ()

output pubblico <?> gradHandle ()