dev.langchain4j.model.embedding

Class OnnxBertBiEncoder

java.lang.Object

dev.langchain4j.model.embedding.OnnxBertBiEncoder

public class OnnxBertBiEncoder
extends Object

Constructor Summary

Constructors

Constructor and Description
OnnxBertBiEncoder(InputStream modelInputStream, URL vocabularyFile, dev.langchain4j.model.embedding.PoolingMode poolingMode)

Method Summary

Modifier and Type	Method and Description
float[]	embed(String text)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

OnnxBertBiEncoder

public OnnxBertBiEncoder(InputStream modelInputStream,
                         URL vocabularyFile,
                         dev.langchain4j.model.embedding.PoolingMode poolingMode)

Method Detail

embed

public float[] embed(String text)