| 158 | } |
| 159 | |
| 160 | array iota(const dim4 &dims, const dim4 &tile_dims, const af::dtype ty) { |
| 161 | af_array out; |
| 162 | AF_THROW(af_iota(&out, dims.ndims(), dims.get(), tile_dims.ndims(), |
| 163 | tile_dims.get(), ty)); |
| 164 | return array(out); |
| 165 | } |
| 166 | |
| 167 | array identity(const dim4 &dims, const af::dtype type) { |
| 168 | af_array res; |