AssignVariableXlaConcatND

kelas akhir publik AssignVariableXlaConcatND

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
buat ( Lingkup lingkup , sumber daya Operand <?>, input Iterable< Operand <T>>, Daftar<Panjang> numConcats, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi AssignVariableXlaConcatND baru.
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.