Menyatukan tensor input di semua dimensi.
Sebuah op yang menggabungkan irisan tensor input berdasarkan atribut num_splits yang diberikan, menghapus padding secara opsional, dan mengembalikan tensor yang digabungkan tanpa padding.
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 | XlaConcatND.Options | Atribut opsional untuk XlaConcatND |
Metode Publik
Output <T> | asOutput () Mengembalikan pegangan simbolis tensor. |
statis <T> XlaConcatND <T> | membuat ( Lingkup lingkup, Iterable < Operan <T >> masukan, Daftar <Panjang> numConcats, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi XlaConcatND baru. |
Output <T> | Output () |
statis XlaConcatND.Options | paddings (Daftar <Panjang> paddings) |
Metode yang Diwarisi
Metode Publik
publik Keluaran <T> asOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static XlaConcatND <T> membuat ( Lingkup lingkup, Iterable < Operan <T >> masukan, Daftar <Panjang> numConcats, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaConcatND baru.
Parameter
cakupan | lingkup saat ini |
---|---|
masukan | Masukkan irisan tensor dalam urutan baris-utama untuk digabungkan di semua dimensi. Semua input harus memiliki bentuk yang sama. } out_arg { nama: "keluaran" deskripsi: < |
numConcats | Jumlah cara untuk menggabungkan per dimensi. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru XlaConcatND
public static XlaConcatND.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. |
---|