공개 최종 클래스 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]
'입력' 모양의 '[s0, s1, ..., s_n]'이 주어지면 출력은 '[s0, s1, ..., s_n / 8]' 모양의 'uint8' 텐서입니다.
상수
끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
출력 < TUint8 > | 출력 () 텐서의 기호 핸들을 반환합니다. |
static <T는 TType을 확장합니다. > CompareAndBitpack | |
출력 < TUint8 > | 출력 () 비트팩 비교. |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "CompareAndBitpack"
공개 방법
공개 출력 < TUint8 > asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 CompareAndBitpack 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <T> 임계값)
새로운 CompareAndBitpack 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
입력 | '임계값' 및 비트팩과 비교할 값입니다. |
한계점 | 비교할 임계값입니다. |
보고
- CompareAndBitpack의 새 인스턴스