My Project
data.h
Go to the documentation of this file.
1 /*******************************************************
2  * Copyright (c) 2014, ArrayFire
3  * All rights reserved.
4  *
5  * This file is distributed under 3-clause BSD license.
6  * The complete license agreement can be obtained at:
7  * http://arrayfire.com/licenses/BSD-3-Clause
8  ********************************************************/
9 
10 #pragma once
11 #include <af/defines.h>
12 
13 #ifdef __cplusplus
14 #include <af/dim4.hpp>
15 #include <af/traits.hpp>
16 namespace af
17 {
18  class array;
19 
30  template<typename T>
31  array constant(T val, const dim4 &dims, const dtype ty=(af_dtype)dtype_traits<T>::ctype);
32 
43  template<typename T>
44  array constant(T val, const dim_t d0, const af_dtype ty=(af_dtype)dtype_traits<T>::ctype);
45 
56  template<typename T>
57  array constant(T val, const dim_t d0, const dim_t d1, const af_dtype ty=(af_dtype)dtype_traits<T>::ctype);
58 
70  template<typename T>
71  array constant(T val, const dim_t d0, const dim_t d1, const dim_t d2, const af_dtype ty=(af_dtype)dtype_traits<T>::ctype);
72 
85  template<typename T>
86  array constant(T val, const dim_t d0, const dim_t d1, const dim_t d2, const dim_t d3, const af_dtype ty=(af_dtype)dtype_traits<T>::ctype);
87 
96  AFAPI array identity(const dim4 &dims, const dtype ty=f32);
97 
106  AFAPI array identity(const dim_t d0, const dtype ty=f32);
107 
117  AFAPI array identity(const dim_t d0, const dim_t d1, const dtype ty=f32);
118 
129  AFAPI array identity(const dim_t d0, const dim_t d1,
130  const dim_t d2, const dtype ty=f32);
131 
143  AFAPI array identity(const dim_t d0, const dim_t d1,
144  const dim_t d2, const dim_t d3, const dtype ty=f32);
145 
155  AFAPI array range(const dim4 &dims, const int seq_dim = -1, const dtype ty=f32);
156 
169  AFAPI array range(const dim_t d0, const dim_t d1 = 1, const dim_t d2 = 1,
170  const dim_t d3 = 1, const int seq_dim = -1, const dtype ty=f32);
171 
181  AFAPI array iota(const dim4 &dims, const dim4 &tile_dims = dim4(1), const dtype ty=f32);
182 
193  AFAPI array diag(const array &in, const int num = 0, const bool extract = true);
194 
205  AFAPI array join(const int dim, const array &first, const array &second);
206 
218  AFAPI array join(const int dim, const array &first, const array &second, const array &third);
219 
232  AFAPI array join(const int dim, const array &first, const array &second,
233  const array &third, const array &fourth);
234 
249  AFAPI array tile(const array &in, const unsigned x, const unsigned y=1,
250  const unsigned z=1, const unsigned w=1);
251 
263  AFAPI array tile(const array &in, const dim4 &dims);
264 
275  AFAPI array reorder(const array& in, const unsigned x,
276  const unsigned y=1, const unsigned z=2, const unsigned w=3);
277 
289  AFAPI array shift(const array& in, const int x, const int y=0, const int z=0, const int w=0);
290 
299  AFAPI array moddims(const array& in, const unsigned ndims, const dim_t * const dims);
300 
308  AFAPI array moddims(const array& in, const dim4& dims);
309 
320  AFAPI array moddims(const array& in, const dim_t d0, const dim_t d1=1, const dim_t d2=1, const dim_t d3=1);
321 
328  AFAPI array flat(const array &in);
329 
337  AFAPI array flip(const array &in, const unsigned dim);
338 
346  AFAPI array lower(const array &in, bool is_unit_diag=false);
347 
355  AFAPI array upper(const array &in, bool is_unit_diag=false);
356 
357 #if AF_API_VERSION >= 31
358 
366  AFAPI array select(const array &cond, const array &a, const array &b);
367 #endif
368 
369 #if AF_API_VERSION >= 31
370 
378  AFAPI array select(const array &cond, const array &a, const double &b);
379 #endif
380 
381 #if AF_API_VERSION >= 31
382 
390  AFAPI array select(const array &cond, const double &a, const array &b);
391 #endif
392 
393 #if AF_API_VERSION >= 31
394 
403  AFAPI void replace(array &a, const array &cond, const array &b);
404 #endif
405 
406 #if AF_API_VERSION >= 31
407 
416  AFAPI void replace(array &a, const array &cond, const double &b);
417 #endif
418 
422 }
423 #endif
424 
425 #ifdef __cplusplus
426 extern "C" {
427 #endif
428 
429 
439  AFAPI af_err af_constant(af_array *arr, const double val, const unsigned ndims, const dim_t * const dims, const af_dtype type);
440 
452  AFAPI af_err af_constant_complex(af_array *arr, const double real, const double imag,
453  const unsigned ndims, const dim_t * const dims, const af_dtype type);
454 
464  AFAPI af_err af_constant_long (af_array *arr, const long long val, const unsigned ndims, const dim_t * const dims);
465 
475  AFAPI af_err af_constant_ulong(af_array *arr, const unsigned long long val, const unsigned ndims, const dim_t * const dims);
489  AFAPI af_err af_range(af_array *out, const unsigned ndims, const dim_t * const dims,
490  const int seq_dim, const af_dtype type);
491 
502  AFAPI af_err af_iota(af_array *out, const unsigned ndims, const dim_t * const dims,
503  const unsigned t_ndims, const dim_t * const tdims, const af_dtype type);
504 
505 
514  AFAPI af_err af_identity(af_array *out, const unsigned ndims, const dim_t * const dims, const af_dtype type);
515 
523  AFAPI af_err af_diag_create(af_array *out, const af_array in, const int num);
524 
532  AFAPI af_err af_diag_extract(af_array *out, const af_array in, const int num);
533 
544  AFAPI af_err af_join(af_array *out, const int dim, const af_array first, const af_array second);
545 
558  AFAPI af_err af_join_many(af_array *out, const int dim, const unsigned n_arrays, const af_array *inputs);
559 
574  AFAPI af_err af_tile(af_array *out, const af_array in,
575  const unsigned x, const unsigned y, const unsigned z, const unsigned w);
576 
587  AFAPI af_err af_reorder(af_array *out, const af_array in,
588  const unsigned x, const unsigned y, const unsigned z, const unsigned w);
589 
600  AFAPI af_err af_shift(af_array *out, const af_array in, const int x, const int y, const int z, const int w);
601 
610  AFAPI af_err af_moddims(af_array *out, const af_array in, const unsigned ndims, const dim_t * const dims);
611 
618  AFAPI af_err af_flat(af_array *out, const af_array in);
619 
627  AFAPI af_err af_flip(af_array *out, const af_array in, const unsigned dim);
628 
636  AFAPI af_err af_lower(af_array *out, const af_array in, bool is_unit_diag);
637 
645  AFAPI af_err af_upper(af_array *out, const af_array in, bool is_unit_diag);
650 #if AF_API_VERSION >= 31
651 
659  AFAPI af_err af_select(af_array *out, const af_array cond, const af_array a, const af_array b);
660 #endif
661 
662 #if AF_API_VERSION >= 31
663 
671  AFAPI af_err af_select_scalar_r(af_array *out, const af_array cond, const af_array a, const double b);
672 #endif
673 
674 #if AF_API_VERSION >= 31
675 
683  AFAPI af_err af_select_scalar_l(af_array *out, const af_array cond, const double a, const af_array b);
684 #endif
685 
686 #if AF_API_VERSION >= 31
687 
696  AFAPI af_err af_replace(af_array a, const af_array cond, const af_array b);
697 #endif
698 
699 #if AF_API_VERSION >= 31
700 
709  AFAPI af_err af_replace_scalar(af_array a, const af_array cond, const double b);
710 #endif
711 
712 #ifdef __cplusplus
713 }
714 #endif
AFAPI af_err af_constant_long(af_array *arr, const long long val, const unsigned ndims, const dim_t *const dims)
AFAPI array join(const int dim, const array &first, const array &second)
Join 2 arrays along dim.
AFAPI array moddims(const array &in, const unsigned ndims, const dim_t *const dims)
AFAPI af_err af_constant_complex(af_array *arr, const double real, const double imag, const unsigned ndims, const dim_t *const dims, const af_dtype type)
AFAPI array lower(const array &in, bool is_unit_diag=false)
Definition: algorithm.h:14
AFAPI array flip(const array &in, const unsigned dim)
AFAPI af_err af_shift(af_array *out, const af_array in, const int x, const int y, const int z, const int w)
AFAPI af_err af_moddims(af_array *out, const af_array in, const unsigned ndims, const dim_t *const dims)
AFAPI af_err af_range(af_array *out, const unsigned ndims, const dim_t *const dims, const int seq_dim, const af_dtype type)
AFAPI af_err af_iota(af_array *out, const unsigned ndims, const dim_t *const dims, const unsigned t_ndims, const dim_t *const tdims, const af_dtype type)
AFAPI af_err af_identity(af_array *out, const unsigned ndims, const dim_t *const dims, const af_dtype type)
AFAPI af_err af_constant_ulong(af_array *arr, const unsigned long long val, const unsigned ndims, const dim_t *const dims)
AFAPI af_err af_replace(af_array a, const af_array cond, const af_array b)
AFAPI af_err af_join_many(af_array *out, const int dim, const unsigned n_arrays, const af_array *inputs)
Join many arrays along dim.
AFAPI af_err af_select(af_array *out, const af_array cond, const af_array a, const af_array b)
AFAPI af_err af_diag_extract(af_array *out, const af_array in, const int num)
AFAPI af_err af_flat(af_array *out, const af_array in)
AFAPI array range(const dim4 &dims, const int seq_dim=-1, const dtype ty=f32)
AFAPI array diag(const array &in, const int num=0, const bool extract=true)
AFAPI array reorder(const array &in, const unsigned x, const unsigned y=1, const unsigned z=2, const unsigned w=3)
af_err
Definition: defines.h:65
AFAPI af_err af_join(af_array *out, const int dim, const af_array first, const af_array second)
Join 2 arrays along dim.
AFAPI af_err af_diag_create(af_array *out, const af_array in, const int num)
AFAPI af_err af_constant(af_array *arr, const double val, const unsigned ndims, const dim_t *const dims, const af_dtype type)
AFAPI array select(const array &cond, const array &a, const array &b)
AFAPI af_err af_upper(af_array *out, const af_array in, bool is_unit_diag)
long long dim_t
Definition: defines.h:50
AFAPI array imag(const array &in)
C++ Interface for getting imaginary part from complex array.
AFAPI array real(const array &in)
C++ Interface for getting real part from complex array.
AFAPI array flat(const array &in)
AFAPI af_err af_select_scalar_r(af_array *out, const af_array cond, const af_array a, const double b)
#define AFAPI
Definition: defines.h:31
AFAPI af_err af_replace_scalar(af_array a, const af_array cond, const double b)
AFAPI af_err af_select_scalar_l(af_array *out, const af_array cond, const double a, const af_array b)
static af::array array(af::dim4 idims, cl_mem buf, af::dtype type, bool retain=false)
Create an af::array object from an OpenCL cl_mem buffer.
Definition: opencl.h:327
AFAPI af_err af_tile(af_array *out, const af_array in, const unsigned x, const unsigned y, const unsigned z, const unsigned w)
array constant(T val, const dim4 &dims, const dtype ty=(af_dtype) dtype_traits< T >::ctype)
AFAPI array tile(const array &in, const unsigned x, const unsigned y=1, const unsigned z=1, const unsigned w=1)
AFAPI af_err af_reorder(af_array *out, const af_array in, const unsigned x, const unsigned y, const unsigned z, const unsigned w)
void * af_array
Definition: defines.h:224
AFAPI void replace(array &a, const array &cond, const array &b)
32-bit floating point values
Definition: defines.h:198
AFAPI af_err af_lower(af_array *out, const af_array in, bool is_unit_diag)
AFAPI array shift(const array &in, const int x, const int y=0, const int z=0, const int w=0)
AFAPI array iota(const dim4 &dims, const dim4 &tile_dims=dim4(1), const dtype ty=f32)
af_dtype
Definition: defines.h:197
AFAPI af_err af_flip(af_array *out, const af_array in, const unsigned dim)
af_dtype dtype
Definition: defines.h:500
AFAPI array upper(const array &in, bool is_unit_diag=false)
AFAPI array identity(const dim4 &dims, const dtype ty=f32)