Package chat.octet.model.components.processor

Class LogitsProcessorList

java.lang.Object

java.util.AbstractCollection<E>

java.util.AbstractList<E>

java.util.ArrayList<LogitsProcessor>

chat.octet.model.components.processor.LogitsProcessorList

All Implemented Interfaces:

LogitsProcessor, Serializable, Cloneable, Iterable<LogitsProcessor>, Collection<LogitsProcessor>, List<LogitsProcessor>, RandomAccess

public final class LogitsProcessorList
extends ArrayList<LogitsProcessor>
implements LogitsProcessor

Stopping criteria list

Author:

William

See Also:

• Serialized Form

Field Summary

Fields inherited from class java.util.AbstractList

modCount

Constructor Summary

Constructors

Constructor	Description
LogitsProcessorList()
LogitsProcessorList(Collection<? extends LogitsProcessor> c)

Method Summary

Modifier and Type	Method	Description
float[]	processor(int[] inputTokenIds, float[] scores, Object... args)	Logits processor

Methods inherited from class java.util.ArrayList

add, add, addAll, addAll, clear, clone, contains, ensureCapacity, equals, forEach, get, hashCode, indexOf, isEmpty, iterator, lastIndexOf, listIterator, listIterator, remove, remove, removeAll, removeIf, removeRange, replaceAll, retainAll, set, size, sort, spliterator, subList, toArray, toArray, trimToSize

Methods inherited from class java.util.AbstractCollection

containsAll, toString

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface java.util.Collection

parallelStream, stream, toArray

Methods inherited from interface java.util.List

containsAll

Constructor Details

LogitsProcessorList

public LogitsProcessorList()

LogitsProcessorList

public LogitsProcessorList(Collection<? extends LogitsProcessor> c)

Method Details

processor

public float[] processor(@Nullable
 int[] inputTokenIds,
 @Nonnull
 float[] scores,
 Object... args)

Description copied from interface: LogitsProcessor

Logits processor

Specified by:

processor in interface LogitsProcessor

Parameters:

inputTokenIds - Indices of input sequence tokens in the vocabulary.

scores - Prediction scores of a language modeling head. These can be logits for each vocabulary.

args - Specific args to a logits processor.

Returns:

float[] The processed prediction scores.