org.tribuo.interop.tensorflow

Class TensorMap

java.lang.Object

org.tribuo.interop.tensorflow.TensorMap

All Implemented Interfaces:

AutoCloseable

public final class TensorMap
extends Object
implements AutoCloseable

A map of names and tensors to feed into a session.

Constructor Summary

Constructors

Constructor and Description
TensorMap(Map<String,org.tensorflow.Tensor> map) Creates a new TensorMap wrapping the supplied map.
TensorMap(String inputName, org.tensorflow.Tensor value) Creates a TensorMap containing the supplied mapping.

Method Summary

Modifier and Type	Method and Description
void	close()
org.tensorflow.Session.Runner	feedInto(org.tensorflow.Session.Runner runner) Feeds the tensors in this FeedDict into the runner.
Map<String,org.tensorflow.Tensor>	getMap() Returns the underlying immutable map.
Optional<org.tensorflow.Tensor>	getTensor(String key) Returns the specified tensor if present.
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

TensorMap

public TensorMap(String inputName,
                 org.tensorflow.Tensor value)

Creates a TensorMap containing the supplied mapping.

Parameters:

inputName - The input name.

value - The tensor value.

TensorMap

public TensorMap(Map<String,org.tensorflow.Tensor> map)

Creates a new TensorMap wrapping the supplied map.

Parameters:

map - A map from strings to tensors.

Method Detail

getMap

public Map<String,org.tensorflow.Tensor> getMap()

Returns the underlying immutable map.

Returns:

The map.

getTensor

public Optional<org.tensorflow.Tensor> getTensor(String key)

Returns the specified tensor if present.

Parameters:

key - The key to lookup.

Returns:

An optional containing the specified tensor if it's in the map, an empty optional otherwise.

feedInto

public org.tensorflow.Session.Runner feedInto(org.tensorflow.Session.Runner runner)

Feeds the tensors in this FeedDict into the runner.

If the session does not have placeholders with the names used in this TensorMap then the Session.Runner will throw IllegalArgumentException.

Parameters:

runner - The session runner.

Returns:

The runner.

close

public void close()

Specified by:

close in interface AutoCloseable

toString

public String toString()

Overrides:

toString in class Object