gistfile1.txt

  val height = 24
  val width = 24
  val depth = 3

  val img = Nd4j.ones(10, depth, height, width)

  val reshapebuilder = new ReshapeVertex(10, 1, 36, 48)

  private def convInit(name: String, in: Int, out: Int, kernel: Array[Int], stride: Array[Int], pad: Array[Int], bias: Double): ConvolutionLayer =
    return new ConvolutionLayer.Builder(kernel, stride, pad).name(name).nIn(in).nOut(out).biasInit(bias).build


  val conf: ComputationGraphConfiguration = new NeuralNetConfiguration.Builder()
    .graphBuilder()
    .addInputs("input")
    .addLayer("L1", convInit("cnn1", depth, 3, Array[Int](1, 1), Array[Int](1, 1), Array[Int](0, 0), 0), "input")
    .addVertex("L2", reshapebuilder, "L1")
    .setOutputs("L2")
    .build

  val net = new ComputationGraph(conf)
  net.init()
  val result: Array[INDArray] = net.output(img)

  println(result)

  println(result(0).shapeInfoToString())