Menggabungkan tensor masukan di semua dimensi.
Sebuah operasi yang menggabungkan irisan tensor masukan berdasarkan atribut num_splits yang diberikan, menghapus bantalan secara opsional, dan menulis tensor gabungan tanpa bantalan 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]]
`jumlah_pemisahan`: [2, 2]
dan `padding`: [1, 1]
`output` yang diharapkan adalah: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Kelas Bersarang
kelas | TetapkanVariableXlaConcatND.Options | Atribut opsional untuk AssignVariableXlaConcatND |
Metode Publik
statis <T> AssignVariableXlaConcatND | |
AssignVariableXlaConcatND.Options statis | bantalan (Daftar<Panjang> bantalan) |
Metode Warisan
Metode Publik
public static AssignVariableXlaConcatND buat ( Lingkup cakupan , sumber daya Operand <?>, input Iterable< Operand <T>>, Daftar<Panjang> numConcats, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi AssignVariableXlaConcatND baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
sumber | Variabel sumber daya untuk tensor masukan gabungan di semua dimensi. } in_arg { nama: "input" deskripsi: < |
nomorConcats | Banyaknya cara untuk menggabungkan per dimensi. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari AssignVariableXlaConcatND
padding AssignVariableXlaConcatND.Options statis publik (padding Daftar<Panjang>)
Parameter
bantalan | Daftar opsional padding kanan per dimensi yang akan dihapus dari tensor gabungan akhir. Bantalan ini tidak boleh melebihi ukuran dimensi hasil penggabungan sebelum pengupasan bantalan. |
---|