Module Autograd.Matrix

Matrices.

type nonrec t = t array array

A matrix.

val init : int -> int -> (int -> int -> t) -> t

Create a matrix with given coefficients.

val rows : 'a array -> int

Number of rows.

val cols : 'a array array -> int

Number of columns.

val row : t -> int -> Vector.t

Get a row.

val coefficients : t -> t list

List of all coefficients.

val ap : t -> Vector.t -> Vector.t

Apply a matrix to a vector.

val transpose : t array array -> t

Transpoe a matrix.