cn.smartjavaai.translation.entity

类 CausalLMOutput

java.lang.Object

cn.smartjavaai.translation.entity.CausalLMOutput

public class CausalLMOutput
extends Object

解码输出对象

作者:

Calvin

构造器概要

构造器

构造器和说明
CausalLMOutput(ai.djl.ndarray.NDArray logits, ai.djl.ndarray.NDList pastKeyValues)

方法概要

限定符和类型	方法和说明
ai.djl.ndarray.NDArray	getLogits()
ai.djl.ndarray.NDList	getPastKeyValuesList()
void	setLogits(ai.djl.ndarray.NDArray logits)

从类继承的方法 java.lang.Object

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

构造器详细资料

CausalLMOutput

public CausalLMOutput(ai.djl.ndarray.NDArray logits,
                      ai.djl.ndarray.NDList pastKeyValues)

方法详细资料

getLogits

public ai.djl.ndarray.NDArray getLogits()

setLogits

public void setLogits(ai.djl.ndarray.NDArray logits)

getPastKeyValuesList

public ai.djl.ndarray.NDList getPastKeyValuesList()