Class Convertor
- java.lang.Object
-
- org.apache.wayang.tensorflow.model.Convertor
-
public class Convertor extends java.lang.Object
-
-
Constructor Summary
Constructors Constructor Description Convertor()
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static org.tensorflow.framework.optimizers.Optimizer
convert(org.tensorflow.Graph graph, Optimizer optimizer)
static org.tensorflow.Operand<org.tensorflow.types.TInt32>
convert(org.tensorflow.op.Ops tf, ArgMax op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Cast op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<org.tensorflow.types.TBool>
convert(org.tensorflow.op.Ops tf, Eq op, org.tensorflow.Operand<?> left, org.tensorflow.Operand<?> right)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Input op)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Mean op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, CrossEntropyLoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, ReLU op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Sigmoid op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Softmax op, org.tensorflow.Operand<?> input)
static org.tensorflow.Operand<?>
convert(org.tensorflow.op.Ops tf, Op op, org.tensorflow.Operand<?>... inputs)
static <T extends org.tensorflow.types.family.TNumber>
org.tensorflow.Operand<T>convertInternal(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<T> input, java.lang.Class<T> tClass)
static org.tensorflow.Tensor
ndArrayToTensor(org.tensorflow.ndarray.NdArray<?> array)
-
-
-
Method Detail
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Op op, org.tensorflow.Operand<?>... inputs)
-
convert
public static org.tensorflow.Operand<org.tensorflow.types.TInt32> convert(org.tensorflow.op.Ops tf, ArgMax op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Cast op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, CrossEntropyLoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)
-
convert
public static org.tensorflow.Operand<org.tensorflow.types.TBool> convert(org.tensorflow.op.Ops tf, Eq op, org.tensorflow.Operand<?> left, org.tensorflow.Operand<?> right)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Input op)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<?> input)
-
convertInternal
public static <T extends org.tensorflow.types.family.TNumber> org.tensorflow.Operand<T> convertInternal(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<T> input, java.lang.Class<T> tClass)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Mean op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, ReLU op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Sigmoid op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Softmax op, org.tensorflow.Operand<?> input)
-
convert
public static org.tensorflow.framework.optimizers.Optimizer convert(org.tensorflow.Graph graph, Optimizer optimizer)
-
ndArrayToTensor
public static org.tensorflow.Tensor ndArrayToTensor(org.tensorflow.ndarray.NdArray<?> array)
-
-