Source file OCamlR_base_stubs2.ml

open OCamlR

let id x = x

module Symbol = struct
  let length = symbol ~generic:true "length"

  let dim = symbol ~generic:true "dim"

  let subset = symbol ~generic:true "["

  let subset2 = symbol ~generic:true "[["

  let rle = symbol ~generic:true "rle"

  let sample = symbol ~generic:true "sample"

  let min = symbol ~generic:true "min"

  let max = symbol ~generic:true "max"
end

let length l = call Symbol.length [ arg id l ]

let dim x =
  call Symbol.dim [ arg id x ]

let subset x i = call Symbol.subset [
    arg id x ;
    arg id i ;
  ]

let subset_ii x i j = call Symbol.subset [
    arg id x ;
    arg id i ;
    arg id j ;
  ]


let subset2_i x i = call Symbol.subset2 [
    arg id x  ;
    arg id i
  ]

let subset2_s = subset2_i

module Matrix = struct
  let subset = subset
  let subset_ii = subset_ii
  let subset2 = subset2_i
end

let rle x = call Symbol.rle [ arg id x ]

let sample x n ?replace ?prob () =
  call Symbol.sample [
    arg id x ;
    arg id n ;
    opt_arg id "replace" replace ;
    opt_arg id "prob" prob
  ]

let min x = call Symbol.min [ arg id x ]
let max x = call Symbol.max [ arg id x ]