CompareAndBitpack

공개 최종 클래스 CompareAndBitpack

'input' 값을 'threshold'와 비교하고 결과 비트를 'uint8'로 압축합니다.

각 비교는 부울 `true`(`입력_값 > 임계값`인 경우)를 반환하고 그렇지 않으면 `false`를 반환합니다.

이 작업은 LSH(Locality-Sensitive-Hashing) 및 코사인 및 'L2' 거리의 해싱 근사치를 사용하는 기타 알고리즘에 유용합니다. 코드는 다음을 통해 입력에서 생성될 수 있습니다:

codebook_size = 50
 codebook_bits = codebook_size * 32
 codebook = tf.get_variable('codebook', [x.shape[-1].value, codebook_bits],
                            dtype=x.dtype,
                            initializer=tf.orthogonal_initializer())
 codes = compare_and_threshold(tf.matmul(x, codebook), threshold=0.)
 codes = tf.bitcast(codes, tf.int32)  # go from uint8 to int32
 # now codes has shape x.shape[:-1] + [codebook_size]
 
참고 : 현재 텐서의 가장 안쪽 차원은 8로 나누어져야 합니다.

'입력' 모양의 '[s0, s1, ..., s_n]'이 주어지면 출력은 '[s0, s1, ..., s_n / 8]' 모양의 'uint8' 텐서입니다.

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 < TUint8 >
출력 ()
텐서의 기호 핸들을 반환합니다.
static <T는 TType을 확장합니다. > CompareAndBitpack
생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <T> 임계값)
새로운 CompareAndBitpack 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 < TUint8 >
출력 ()
비트팩 비교.

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "CompareAndBitpack"

공개 방법

공개 출력 < TUint8 > asOutput ()

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

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

공개 정적 CompareAndBitpack 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <T> 임계값)

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

매개변수
범위 현재 범위
입력 '임계값' 및 비트팩과 비교할 값입니다.
한계점 비교할 임계값입니다.
보고
  • CompareAndBitpack의 새 인스턴스

공개 출력 < TUint8 > 출력 ()

비트팩 비교.