BandPart

공개 최종 수업 BandPart

각 가장 안쪽 행렬의 중앙 밴드 외부에 있는 모든 항목을 0으로 설정하는 텐서를 복사합니다.

`밴드` 부분은 다음과 같이 계산됩니다. `입력`에 `k` 차원 `[I, J, K, ..., M, N]`이 있다고 가정하면 출력은 동일한 모양의 텐서입니다.

`밴드[i, j, k, ..., m, n] = in_band(m, n) * 입력[i, j, k, ..., m, n]`.

표시기 기능

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

예를 들어:

# 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]]
 
유용한 특수 사례:
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.
 

상수

OP_NAME TensorFlow 핵심 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
출력 <T>
밴드 ()
입력과 동일한 모양의 'k' 텐서 순위를 지정합니다.
static <T는 TType을 확장하고, U는 TNumber를 확장합니다. > BandPart <T>
생성 ( 범위 범위 , 피연산자 <T> 입력, 피연산자 <U> numLower, 피연산자 <U> numUpper)
새 BandPart 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 핵심 엔진에서 알려진 이 작업의 이름

상수 값: "MatrixBandPart"

공개 방법

공개 출력 <T> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

공개 출력 <T> 밴드 ()

입력과 동일한 모양의 'k' 텐서 순위를 지정합니다. 추출된 줄무늬 텐서.

공개 정적 BandPart <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <U> numLower, 피연산자 <U> numUpper)

새 BandPart 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
입력 `k` 텐서 순위.
숫자낮음 0차원 텐서. 유지할 하부대각선의 수입니다. 음수이면 아래쪽 삼각형 전체를 유지합니다.
숫자상한 0차원 텐서. 유지할 상부대각선의 수입니다. 음수인 경우 전체 위쪽 삼각형을 유지합니다.
보고
  • BandPart의 새 인스턴스