Op("sizeAt") {
        javaPackage = "org.nd4j.linalg.api.ops.impl.shape"
        Input(NDARRAY, "in") { description = "Input variable" }
        Arg(INT, "dimension") { description = "Dimension to get size of" }
        Output(NUMERIC, "output"){ description = "Scalar INDArray  for size at specified variable" }