Function identity

src/api/cpp/data.cpp:167–171

}

array identity(const dim4 &dims, const af::dtype type) {
    af_array res;
    AF_THROW(af_identity(&res, dims.ndims(), dims.get(), type));
    return array(res);
}

array identity(const dim_t d0, const af::dtype ty) {
    return identity(dim4(d0), ty);