uninitVariable

create new variable with uninitialized array of shape on CPU/CUDA

uninitVariable
(
T
size_t dim
)
(
uint[dim] shape
,
bool requiresGrad = false
)

Examples

1 import std.stdio;
2 import numir;
3 import mir.ndslice;
4 
5 auto x = numir.zeros(2, 3, 4).universal;
6 auto y = uninitVariable!float([2, 3, 4]);
7 assert(x.strides == y.strides);

Meta