Beware: This is a work in progress on a highly experimental prototype. Do not expect backward compatibility, performance nor correctness.
Tensority is an experience in designing a library for strongly-typed multidimensional arrays and tensors manipulation. Tensority aims to cover three levels of compile-time safety:
-
tensor order level:
- the type of multidimensional arrays should distinguish between arrays of different dimensions
- the type of tensor should moreover distinguish between an vector and 1-form,
i.e between
(1 + 0)tensor and(0 + 1)tensors
-
dimension level:
- adding two vectors of different dimensions should be a type error
-
index level
- trying to access the
k+1th elements of an array of sizekshould be an error
- trying to access the
Each level of safety restricts the number of functions implementable using tensority safe interface. However, non-trivial functions can still be implemented using this safe interface.
With the included ppx extension, tensority looks like:
open Tensority
open Multidim_array
open Shape
(* Multidimensional array literals *)
let array4 = [%array 4
[
[1, 2; 3, 4], [5, 6; 7, 8], [9, 10; 11, 12];
[1, 2; 8, 9], [9, 1; 2, 3], [19, 12; 14, 17]
]
];;
(* Definition using function *)
let array = init_sh [502103s] (function [k] -> Nat.to_int k)
(* accessing an element *)
let one = array4.( 1i, 0i, 0i, 0i ) ;;
(* accessing an element with an out-of-bound type error *)
let one = array4.( 1i, 5i, 0i, 1i ) ;;
(* element assignment *)
array4.(0i,0i,0i,0i) <- 0;;
(* slicing *)
let matrix = array4.!( All, All, 1j, 1j );;
(* matrix is [9, 1; 2, 3] *)
(* slice assignment *)
let row = [%array (2,3) ];;
matrix.!(All, 0j) <- row;;
(* matrix is now [2, 3; 2, 3] *)
(* range slice *)
let array' = array.!([%range 25 50 ~by:5])
(* or *)
let array' = array.!( 25 #-># 50 ## 5 )Note that most of the ppx transformations are local, without rewriter the previous example would become
open Tensority
open Multidim_array
open Shape
open Nat_defs
(* Multidimensional array literals *)
let array4 = Unsafe.create [ _2; _2; _3; _2 ]
[| 1; 2; 3; 4; 5; 6; 7; 8; 9; 10; 11; 12;
1; 2; 8; 9; 9; 1; 2; 3; 19; 12; 14; 17
|]
(* Definition using function *)
let array = init_sh Size.[nat _5 @ _0 @ _2 @ _1 @ _0 @ _3n]
(function [k] -> Nat.to_int k)
(* accessing an element *)
let one = array4.%( [ _1i; _0i; _0i; _0i] ) ;;
(* accessing an element with an out-of-bound type error *)
let one = array4.%([ _1i; _5i; _0i; _1i ] ) ;;
(* element assignment *)
array4.%([ _0i; _0i; _0i; _0i ]) <- 0;;
(* slicing *)
let matrix = array4.%[[ All; All; Elt _1i; Elt _1i ]];;
(* matrix is [9, 1; 2, 3] *)
(* slice assignment *)
let row = Unsafe.create [_2] [| 2; 3 |];;
matrix.%[[ All; Elt _0i]] <- row
(* matrix is now [2, 3; 2, 3] *)
(* range slice *)
let r= Range.create ~by: 5 ~start:Indices.(nat _2 @ 5n)
~stop:Indices.(nat _5 @ 0n) ~len:_10
let array' = array.%[[Range r]]