BandPart

lớp cuối cùng công khai BandPart

Sao chép một tenxơ đặt mọi thứ bên ngoài dải trung tâm trong mỗi ma trận trong cùng về 0.

Phần `band` được tính như sau: Giả sử `input` có `k` kích thước `[I, J, K, ..., M, N]`, thì đầu ra là một tensor có cùng hình dạng trong đó

`band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n]`.

Chức năng chỉ báo

`in_band(m, n) = (num_low < 0 || (mn) <= num_low)) && (num_upper < 0 || (nm) <= num_upper)`.

Ví dụ:

# 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]]
 
Các trường hợp đặc biệt hữu ích:
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.
 

Hằng số

Sợi dây OP_NAME Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Phương pháp công cộng

Đầu ra <T>
asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầu ra <T>
ban nhạc ()
Xếp hạng tensor `k` có cùng hình dạng với đầu vào.
static <T mở rộng TType , U mở rộng TNumber > BandPart <T>
tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Toán hạng <U> numLower, Toán hạng <U> numUpper)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động BandPart mới.

Phương pháp kế thừa

Hằng số

Chuỗi cuối cùng tĩnh công khai OP_NAME

Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Giá trị không đổi: "MatrixBandPart"

Phương pháp công cộng

Đầu ra công khai <T> asOutput ()

Trả về tay cầm tượng trưng của tensor.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

Đầu ra công khai băng tần <T> ()

Xếp hạng tensor `k` có cùng hình dạng với đầu vào. Tensor dải được trích xuất.

public static BandPart <T> tạo ( Phạm vi phạm vi , đầu vào Toán hạng <T>, Toán hạng <U> numLower, Toán hạng <U> numUpper)

Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động BandPart mới.

Thông số
phạm vi phạm vi hiện tại
đầu vào Xếp hạng tenxơ `k`.
numLower Tenxơ 0-D. Số lượng đường chéo phụ cần giữ. Nếu âm, giữ nguyên toàn bộ tam giác phía dưới.
numUpper Tenxơ 0-D. Số lượng đường chéo cần giữ. Nếu âm, giữ nguyên toàn bộ tam giác trên.
Trả lại
  • một phiên bản mới của BandPart