XlaSplitND

kelas akhir publik XlaSplitND

Membagi tensor masukan di semua dimensi.

Sebuah operasi yang mengiris tensor masukan berdasarkan atribut num_splits yang diberikan, memasukkan irisan secara opsional, dan mengembalikan irisan tersebut. Irisan dikembalikan dalam urutan baris-utama.

Operasi ini dapat dihasilkan melalui jembatan TPU.

Misalnya, dengan tensor `input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
`jumlah_pemisahan`:
[2, 2]
 
dan `padding`:
[1, 1]
 
`output` yang diharapkan adalah:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Kelas Bersarang

kelas XlaSplitND.Opsi Atribut opsional untuk XlaSplitND

Metode Publik

statis <T> XlaSplitND <T>
buat ( Lingkup lingkup , masukan Operan <T>, Panjang N, Daftar<Panjang> numSplits, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSplitND baru.
Iterator< Operan <T>>
Daftar< Keluaran <T>>
XlaSplitND.Options statis
bantalan (Daftar<Panjang> bantalan)

Metode Warisan

Metode Publik

public static XlaSplitND <T> buat ( Lingkup lingkup , masukan Operan <T>, Long N, Daftar<Panjang> numSplits, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaSplitND baru.

Parameter
cakupan ruang lingkup saat ini
masukan Masukkan tensor untuk membagi seluruh dimensi. } out_arg { nama: "keluaran" deskripsi: <
nomorSplit Banyaknya cara untuk membagi per dimensi. Dimensi bentuk harus dapat dibagi rata.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru XlaSplitND

Iterator publik< Operan <T>> iterator ()

Daftar publik< Keluaran <T>> keluaran ()

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

Parameter
bantalan Daftar opsional bantalan kanan per dimensi tensor masukan yang akan diterapkan sebelum pemisahan. Ini dapat digunakan untuk membuat suatu dimensi dapat dibagi rata.