Struct InfoNCELoss 

pub struct InfoNCELoss { /* private fields */ }

InfoNCE contrastive loss for representation learning

Implementations

impl InfoNCELoss

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl InfoNCELoss

pub fn into_reference( val: InfoNCELoss, env: Env, ) -> Result<Reference<InfoNCELoss>>

pub fn into_instance(self, env: Env) -> Result<ClassInstance<InfoNCELoss>>

impl InfoNCELoss

pub fn new(temperature: f64) -> Self

Create a new InfoNCE loss instance

Arguments

• temperature - Temperature parameter for softmax (typically 0.07-0.1)

pub fn compute( &self, anchor: Float32Array, positive: Float32Array, negatives: Vec<Float32Array>, ) -> f64

Compute InfoNCE loss

Arguments

• anchor - Anchor embedding
• positive - Positive example embedding
• negatives - Array of negative example embeddings

pub fn compute_with_gradients( &self, anchor: Float32Array, positive: Float32Array, negatives: Vec<Float32Array>, ) -> LossWithGradients

Compute InfoNCE loss with gradients

Returns an object with loss and gradients fields

pub fn temperature(&self) -> f64

Get the temperature

Trait Implementations

impl FromNapiMutRef for InfoNCELoss

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for InfoNCELoss

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &InfoNCELoss

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut InfoNCELoss

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for InfoNCELoss

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for InfoNCELoss

unsafe fn to_napi_value(env: napi_env, val: InfoNCELoss) -> Result<napi_value>

Safety

impl TypeName for &InfoNCELoss

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut InfoNCELoss

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for InfoNCELoss

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &InfoNCELoss

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut InfoNCELoss

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety