ch.epfl.leb.defcon.utils

Class GraphBuilder

java.lang.Object

ch.epfl.leb.defcon.utils.GraphBuilder

public class GraphBuilder
extends Object

Appends various operations to TensorFlow graphs.

Author:

Baptiste Ottino

See Also:

Output | TensorFlow, Graph | TensorFlow

Constructor Summary

Constructors

Constructor and Description
GraphBuilder()

Method Summary

Modifier and Type	Method and Description
static org.tensorflow.Output<Integer>	constant(org.tensorflow.Graph g, String name, int value) Builds an Output tensor with a single scalar value.
static <T> org.tensorflow.Output<T>	constant(org.tensorflow.Graph g, String name, Object value, Class<T> type) Builds an Output tensor with a single scalar value.
static <T> org.tensorflow.Output<T>	expandDims(org.tensorflow.Graph g, String name, org.tensorflow.Output<T> input, org.tensorflow.Output<Integer> dim) Adds a dimension to a TensorFlow Output tensor.

Methods inherited from class java.lang.Object

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

Constructor Detail

GraphBuilder

public GraphBuilder()

Method Detail

constant

public static <T> org.tensorflow.Output<T> constant(org.tensorflow.Graph g,
                                                    String name,
                                                    Object value,
                                                    Class<T> type)

Builds an Output tensor with a single scalar value.

Type Parameters:

T -

Parameters:

g - The TensorFlow graph to modify.

name - The full name of the operation.

value - The value output by the operation.

type - The output datatype.

Returns:

Symbolic handle to the tensor produced by the appended operation.

constant

public static org.tensorflow.Output<Integer> constant(org.tensorflow.Graph g,
                                                      String name,
                                                      int value)

Builds an Output tensor with a single scalar value.

Parameters:

g - The TensorFlow graph to modify.

name - The full name of the operation.

value - The value output by the operation.

Returns:

Symbolic handle to the tensor produced by the appended operation.

expandDims

public static <T> org.tensorflow.Output<T> expandDims(org.tensorflow.Graph g,
                                                      String name,
                                                      org.tensorflow.Output<T> input,
                                                      org.tensorflow.Output<Integer> dim)

Adds a dimension to a TensorFlow Output tensor. The dimension as added at the position "dim." (with numpy/tensorflow syntax: 0 for the first position, -1 in the end)

Type Parameters:

T -

Parameters:

g - The TensorFlow graph to modify.

name - The full name of the operation.

input - The input operation to expand.

dim -

Returns:

Symbolic handle to the tensor produced by the appended operation.