All Downloads are FREE. Search and download functionalities are using the official Maven repository.

one.empty3.tfj.HelloTensorFlow Maven / Gradle / Ivy

There is a newer version: 2024.5.10
Show newest version
/*
package one.empty3.tfj;

import org.tensorflow.ConcreteFunction;
import org.tensorflow.Signature;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;
import org.tensorflow.op.Ops;
import org.tensorflow.op.core.Placeholder;
import org.tensorflow.op.math.Add;
import org.tensorflow.types.TInt32;

public class HelloTensorFlow {

  public static void main(String[] args) throws Exception {
    Logger.getAnonymousLogger().log(Level.INFO, "Hello TensorFlow " + TensorFlow.version());

    try (ConcreteFunction dbl = ConcreteFunction.create(HelloTensorFlow::dbl);
        TInt32 x = TInt32.scalarOf(10);
        Tensor dblX = dbl.call(x)) {
      Logger.getAnonymousLogger().log(Level.INFO, x.getInt() + " doubled is " + ((TInt32)dblX).getInt());
    }
  }

  private static Signature dbl(Ops tf) {
    Placeholder x = tf.placeholder(TInt32.class);
    Add dblX = tf.math.add(x, x);
    return Signature.builder().input("x", x).output("dbl", dblX).build();
  }
}*/




© 2015 - 2024 Weber Informatics LLC | Privacy Policy