clj-djl.ndarray
acos
(acos ndarray)Returns the inverse trigonometric cosine of this NDArray element-wise.
eye
(eye manager rows)(eye manager rows k)(eye manager rows cols k)(eye manager rows cols k datatype-)(eye manager rows cols k datatype- device)full
(full manager shape- data)(full manager shape- data datatype-)(full manager shape- data datatype- device)get-gradient
(get-gradient ndarray)Returns the gradient NDArray attached to this NDArray.
linspace
(linspace manager start stop n)(linspace manager start stop n endpoint)(linspace manager start stop n endpoint device)ones
(ones manager shape-)(ones manager shape- datatype-)(ones manager shape- datatype- device)random-multinomial
(random-multinomial ndmanager n ndarray)(random-multinomial ndmanager n ndarray shape-)Draw samples from a multinomial distribution.
random-normal
(random-normal manager shape-)(random-normal manager loc scale shape-)(random-normal manager loc scale shape- datatype-)(random-normal manager loc scale shape- datatype- device)random-uniform
(random-uniform manager low high shape-)(random-uniform manager low high shape- datatype-)(random-uniform manager low high shape- datatype- device)set-requires-gradient
(set-requires-gradient ndarray requires-grad)Attaches a gradient NDArray to this NDArray and marks it so GradientCollector.backward(NDArray) can compute the gradient with respect to it.
to-type
(to-type ndarray datatype- copy)convert ndarray to data-type, available options are: “int8” “uint8” “int32” “int64” “float16” “float32” “float64” “boolean” “string” “unknown”
zeros
(zeros manager shape-)(zeros manager shape- datatype-)(zeros manager shape- datatype- device)