Package dev.langchain4j.model.embedding

Class OnnxBertBiEncoder

java.lang.Object

dev.langchain4j.model.embedding.OnnxBertBiEncoder

public class OnnxBertBiEncoder
extends Object

Constructor Summary

Constructors

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

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Details

OnnxBertBiEncoder

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

Method Details

embed

public float[] embed(String text)