Menyatukan tensor input di semua dimensi.
Sebuah op yang menggabungkan mengiris tensor input berdasarkan atribut num_splits yang diberikan, menghapus padding secara opsional, dan menulis tensor yang digabungkan tanpa padding ke variabel sumber daya.
Operasi ini dapat dihasilkan melalui jembatan TPU.
Misalnya, dengan `tensor input`:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
` num_splits`: [2, 2]
dan `paddings`: [1, 1]
yang diharapkan` outputs` adalah: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Kelas Bersarang
kelas | AssignVariableXlaConcatND.Options | Atribut opsional untuk AssignVariableXlaConcatND |
Metode Publik
statis <T> AssignVariableXlaConcatND | membuat ( Lingkup lingkup, Operan <?> sumber daya, Iterable < Operan <T >> masukan, Daftar <Panjang> numConcats, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi AssignVariableXlaConcatND baru. |
statis AssignVariableXlaConcatND.Options | paddings (Daftar <Panjang> paddings) |
Metode yang Diwarisi
Metode Publik
public static AssignVariableXlaConcatND membuat ( Scope lingkup, Operan <?> sumber daya, Iterable < Operan <T >> masukan, Daftar <Panjang> numConcats, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi AssignVariableXlaConcatND baru.
Parameter
cakupan | lingkup saat ini |
---|---|
sumber | Variabel resource untuk tensor input gabungan di semua dimensi. } in_arg { nama: "input" deskripsi: < |
numConcats | Jumlah cara untuk menggabungkan per dimensi. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari AssignVariableXlaConcatND
public static AssignVariableXlaConcatND.Options paddings (Daftar <Panjang> paddings)
Parameter
bantalan | Daftar opsional padding kanan per dimensi untuk dilepas dari tensor gabungan akhir. Padding ini tidak boleh melebihi ukuran dimensi dari hasil penggabungan sebelum stripping padding. |
---|