Class NamedTensors

java.lang.Object

smile.onnx.genai.NamedTensors

All Implemented Interfaces:

AutoCloseable

public final class NamedTensors
extends Object
implements AutoCloseable

Named collection of tensors matching model input names.

Method Summary

Modifier and Type	Method	Description
void	close()
long	count()	Returns the number of named tensors.
static NamedTensors	create()	Creates an empty named-tensor container.
NamedTensors	delete(String name)	Deletes a named tensor entry.
Tensor	get(String name)	Returns a copy of the named tensor.
String[]	names()	Returns the tensor names.
NamedTensors	set(String name, Tensor tensor)	Sets a named tensor (takes a reference; the tensor must outlive use).

Methods inherited from class Object

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

Method Details

create

public static NamedTensors create()

Creates an empty named-tensor container.

Returns:

a new container owned by the caller.

set

public NamedTensors set(String name,
 Tensor tensor)

Sets a named tensor (takes a reference; the tensor must outlive use).

Parameters:

name - input name.

tensor - tensor value.

Returns:

this for chaining.

get

public Tensor get(String name)

Returns a copy of the named tensor.

Parameters:

name - input name.

Returns:

tensor owned by the caller.

delete

public NamedTensors delete(String name)

Deletes a named tensor entry.

Parameters:

name - input name.

Returns:

this for chaining.

count

public long count()

Returns the number of named tensors.

Returns:

count.

names

public String[] names()

Returns the tensor names.

Returns:

names.

close

public void close()

Specified by:

close in interface AutoCloseable