Membuat TensorArray untuk menyimpan beberapa gradien nilai dalam pegangan yang diberikan.
Mirip dengan TensorArrayGradV3. Namun itu menciptakan akumulator dengan bentuk yang diperluas dibandingkan dengan input TensorArray yang gradiennya sedang dihitung. Ini memungkinkan beberapa gradien untuk TensorArray yang sama dihitung menggunakan akumulator yang sama.
Metode Publik
statis TensorArrayGradWithShape | |
Output <Lampung> | flowOut () |
Output <?> | gradHandle () |
Metode yang Diwarisi
Metode Publik
public static TensorArrayGradWithShape membuat ( Scope lingkup, Operan <?> handle, Operan <Lampung> flowIn, Operan <Integer> shapeToPrepend, sumber String)
Metode pabrik untuk membuat kelas yang membungkus operasi TensorArrayGradWithShape baru.
Parameter
cakupan | lingkup saat ini |
---|---|
menangani | Pegangan untuk meneruskan TensorArray. |
aliranIn | Skalar pelampung yang memberlakukan rantai operasi yang tepat. |
bentukToPrepend | Vektor int32 mewakili bentuk. Elemen dalam akumulator gradien akan memiliki bentuk yang merupakan nilai shape_to_prepend yang digabungkan dengan bentuk elemen di TensorArray yang sesuai dengan pegangan input. |
sumber | String sumber gradien, digunakan untuk memutuskan gradien TensorArray mana yang akan dikembalikan. |
Kembali
- contoh baru dari TensorArrayGradWithShape