XlaConcatND

kelas akhir publik XlaConcatND

Menggabungkan tensor masukan di semua dimensi.

Sebuah operasi yang menggabungkan irisan tensor masukan berdasarkan atribut num_splits yang diberikan, menghapus bantalan secara opsional, dan mengembalikan tensor gabungan tanpa bantalan.

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]
 
`outputs` yang diharapkan adalah:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

Kelas Bersarang

kelas XlaConcatND.Opsi Atribut opsional untuk XlaConcatND

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
statis <T> XlaConcatND <T>
buat ( Lingkup lingkup, input Iterable< Operand <T>>, Daftar<Panjang> numConcats, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaConcatND baru.
Keluaran <T>
XlaConcatND.Options statis
bantalan (Daftar<Panjang> bantalan)

Metode Warisan

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static XlaConcatND <T> buat ( Lingkup lingkup, input Iterable< Operand <T>>, Daftar<Panjang> numConcats, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaConcatND baru.

Parameter
cakupan ruang lingkup saat ini
masukan Masukkan irisan tensor dalam urutan baris-utama untuk digabungkan di semua dimensi. Semua masukan harus memiliki bentuk yang sama. } out_arg { nama: "keluaran" deskripsi: <
nomorConcats Banyaknya cara untuk menggabungkan per dimensi.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru XlaConcatND

Keluaran publik <T> keluaran ()

padding XlaConcatND.Options statis publik (padding Daftar<Panjang>)

Parameter
bantalan Daftar opsional padding kanan per dimensi yang akan dihapus dari tensor gabungan akhir. Padding ini tidak boleh melebihi ukuran dimensi hasil penggabungan sebelum melakukan pengupasan padding.