Embedding

공개 추상 클래스 임베딩

지정된 삽입기 헤드의 임베딩을 나타냅니다. 일반적으로 임베딩 태스크에 사용됩니다.

두 'floatEmbedding' 및 'quantizedEmbedding' 중 하나만 임베딩이 스칼라 양자화를 수행하도록 구성되었는지 여부에 따라 데이터가 포함됩니다.

공개 생성자

Embedding()

공개 메서드

정적 임베딩
create(float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, 선택사항<String> headName)
Embedding 인스턴스를 생성합니다.
정적 임베딩
createFromProto(EmbeddingsProto.Embedding proto)
ERROR(/EmbeddingsProto.Embedding) protobuf 메시지에서 Embedding 객체를 만듭니다.
추상 float[]
floatEmbedding()
부동 소수점 임베딩
추상 정수
headIndex()
이러한 항목이 참조하는 임베딩 헤드의 색인입니다.
추상 선택사항<String>
headName()
상응하는 텐서 메타데이터 이름인 임베더 헤드의 이름(선택사항)입니다.
추상 byte[]
quantizedEmbedding()이 있습니다.
양자화 임베딩

상속된 메서드

공개 생성자

public 임베딩 ()

공개 메서드

public static 임베딩 create (float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, 선택사항<String> headName)

Embedding 인스턴스를 생성합니다.

매개변수
floatEmbedding 부동 소수점 임베딩을 사용하면
quantizedEmbedding 양자화 임베딩에 포함됩니다.
headIndex 삽입기 헤드의 색인입니다.
headName 삽입기 헤드의 이름(선택사항)입니다.

public static 임베딩 createFromProto (EmbeddingsProto.Embedding proto)

ERROR(/EmbeddingsProto.Embedding) protobuf 메시지에서 Embedding 객체를 만듭니다.

매개변수
proto 변환할 ERROR(/EmbeddingsProto.Embedding) protobuf 메시지

public 추상 float[] floatEmbedding ()

부동 소수점 임베딩

임베드가 스칼라 양자화를 수행하도록 구성된 경우 비어 있습니다.

public 추상화 int headIndex ()

이러한 항목이 참조하는 임베딩 헤드의 색인입니다. 이는 다중 헤드 모델에 유용합니다.

public summary 선택사항<String> headName ()

상응하는 텐서 메타데이터 이름인 임베더 헤드의 이름(선택사항)입니다.

public 추상화 byte[] quantizedEmbedding ()

양자화 임베딩

임베드가 스칼라 양자화를 수행하도록 구성되지 않은 경우 비어 있습니다.