package caisar
Install
dune-project
Dependency
Authors
Maintainers
Sources
sha256=05024c094f68b82873f2c99c89d4f196049ac63b7d1d4f68ae1a1e3b08de7342
sha512=a26c724a19fca7c22a000367d1cd79c1e0474f373bf7265449928e55275ac44103190536dc8c76f5ac00a2a1897c3bd2ee06bb6f22140165079b72a27011e6df
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).