Salin pengaturan tensor segala sesuatu di luar pita pusat di setiap matriks terdalam ke nol.
Bagian `band` dihitung sebagai berikut: Asumsikan `input` memiliki dimensi `k` `[I, J, K, ..., M, N]`, maka outputnya berupa tensor dengan bentuk yang sama dimana
`band[i, j, k, ..., m, n] = in_band(m, n) * masukan[i, j, k, ..., m, n]`.
Fungsi indikator
`in_band(m, n) = (angka_bawah < 0 || (mn) <= angka_bawah)) && (angka_atas < 0 || (nm) <= angka_atas)`.
Misalnya:
# if 'input' is [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[-2, -1, 0, 1]
[-3, -2, -1, 0]],
tf.matrix_band_part(input, 1, -1) ==> [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[ 0, -1, 0, 1]
[ 0, 0, -1, 0]],
tf.matrix_band_part(input, 2, 1) ==> [[ 0, 1, 0, 0]
[-1, 0, 1, 0]
[-2, -1, 0, 1]
[ 0, -2, -1, 0]]
Kasus khusus yang berguna: tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
tf.matrix_band_part(input, 0, 0) ==> Diagonal.
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
Keluaran <T> | pita () Beri peringkat tensor `k` dengan bentuk yang sama dengan masukan. |
statis <T memperluas TType , U memperluas TNumber > BandPart <T> |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
keluaran publik <T> pita ()
Beri peringkat tensor `k` dengan bentuk yang sama dengan masukan. Tensor berpita yang diekstraksi.
public static BandPart <T> buat ( Lingkup lingkup , masukan Operan <T>, Operan <U> numLower, Operan <U> numUpper)
Metode pabrik untuk membuat kelas yang membungkus operasi BandPart baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
masukan | Beri peringkat tensor `k`. |
angkaLebih Rendah | Tensor 0-D. Jumlah subdiagonal yang harus dipertahankan. Jika negatif, pertahankan seluruh segitiga bawah. |
nomor Atas | Tensor 0-D. Jumlah superdiagonal yang harus dipertahankan. Jika negatif, pertahankan seluruh segitiga atas. |
Kembali
- contoh baru dari BandPart