package caisar
Install
dune-project
Dependency
Authors
Maintainers
Sources
sha256=58ba1e38721795b306c860b56aaeba971be586cd55fb96e3ec8af72dd005101b
sha512=f1b3b9899660745598cebe7ecb52a39e9e16dcb7352381ea75a80d2afa988437130c00bf66355991421d4cb3dc06b02c185f7d4bdcc1c86dfcde8084bd01a654
doc/caisar.nir/Nir/Gentensor/index.html
Module Nir.GentensorSource
type t = | Float of (float, Bigarray.float64_elt) Tensor.t| Int8 of (int, Bigarray.int8_signed_elt) Tensor.t| UInt8 of (int, Bigarray.int8_unsigned_elt) Tensor.t| Int32 of (int32, Bigarray.int32_elt) Tensor.t| Int64 of (int64, Bigarray.int64_elt) Tensor.t
create_const_float shape v returns a tensor of shape shape where each value is initialized to v.
of_float_array a shape returns a Tensor with data contained in l and shape shape. If no shape is given, the resulting tensor shape is 1-dimensional, equals to the length of l.
of_int64_array a shape returns a Tensor with data contained in l and shape shape. If no shape is given, the resulting tensor shape is 1-dimensional, equals to the length of l.
of_float_matrix trans m returns a Tensor with data contained in m. If trans is true (default: false), then m is transposed, i.e., if m had shape (n,k), then the result will have shape (k,n).
of_float_matrix m returns a Tensor with data contained in m but in reversed shape. In other words, if m had shape (n,k), then the result has shape (k,n).