Tensors ยท TensorFlow Scala
TensorFlow Scala - Documentation