A high-performance general-purpose compute library
data.h File Reference
#include <af/defines.h>
#include <af/dim4.hpp>
#include <af/traits.hpp>

Go to the source code of this file.

Namespaces

namespace  af
 

Functions

template<typename T >
array constant (T val, const dim4 &dims, const dtype ty=(af_dtype) dtype_traits< T >::ctype)
 C++ Interface to generate an array with elements set to a specified value. More...
 
template<typename T >
array constant (T val, const dim_t d0, const af_dtype ty=(af_dtype) dtype_traits< T >::ctype)
 C++ Interface to generate a 1-D array with elements set to a specified value. More...
 
template<typename T >
array constant (T val, const dim_t d0, const dim_t d1, const af_dtype ty=(af_dtype) dtype_traits< T >::ctype)
 C++ Interface to generate a 2-D array with elements set to a specified value. More...
 
template<typename T >
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)
 C++ Interface to generate a 3-D array with elements set to a specified value. More...
 
template<typename T >
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)
 C++ Interface to generate a 4-D array with elements set to a specified value. More...
 
AFAPI array identity (const dim4 &dims, const dtype ty=f32)
 C++ Interface to generate an identity array. More...
 
AFAPI array identity (const dim_t d0, const dtype ty=f32)
 C++ Interface to generate a 1-D identity array. More...
 
AFAPI array identity (const dim_t d0, const dim_t d1, const dtype ty=f32)
 C++ Interface to generate a 2-D identity array. More...
 
AFAPI array identity (const dim_t d0, const dim_t d1, const dim_t d2, const dtype ty=f32)
 C++ Interface to generate a 3-D identity array. More...
 
AFAPI array identity (const dim_t d0, const dim_t d1, const dim_t d2, const dim_t d3, const dtype ty=f32)
 C++ Interface to generate a 4-D identity array. More...
 
AFAPI array range (const dim4 &dims, const int seq_dim=-1, const dtype ty=f32)
 C++ Interface to generate an array with [0, n-1] values along the seq_dim dimension and tiled across other dimensions of shape dim4. More...
 
AFAPI array range (const dim_t d0, const dim_t d1=1, const dim_t d2=1, const dim_t d3=1, const int seq_dim=-1, const dtype ty=f32)
 C++ Interface to generate an array with [0, n-1] values along the seq_dim dimension and tiled across other dimensions described by dimension parameters. More...
 
AFAPI array iota (const dim4 &dims, const dim4 &tile_dims=dim4(1), const dtype ty=f32)
 C++ Interface to generate an array with [0, n-1] values modified to specified dimensions and tiling. More...
 
AFAPI array diag (const array &in, const int num=0, const bool extract=true)
 C++ Interface to extract the diagonal from an array. More...
 
AFAPI array join (const int dim, const array &first, const array &second)
 C++ Interface to join 2 arrays along a dimension. More...
 
AFAPI array join (const int dim, const array &first, const array &second, const array &third)
 C++ Interface to join 3 arrays along a dimension. More...
 
AFAPI array join (const int dim, const array &first, const array &second, const array &third, const array &fourth)
 C++ Interface to join 4 arrays along a dimension. More...
 
AFAPI array tile (const array &in, const unsigned x, const unsigned y=1, const unsigned z=1, const unsigned w=1)
 C++ Interface to generate a tiled array. More...
 
AFAPI array tile (const array &in, const dim4 &dims)
 C++ Interface to generate a tiled array. More...
 
AFAPI array reorder (const array &in, const unsigned x, const unsigned y=1, const unsigned z=2, const unsigned w=3)
 C++ Interface to reorder an array. More...
 
AFAPI array shift (const array &in, const int x, const int y=0, const int z=0, const int w=0)
 C++ Interface to shift an array. More...
 
AFAPI array moddims (const array &in, const dim4 &dims)
 C++ Interface to modify the dimensions of an input array to a specified shape. More...
 
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)
 C++ Interface to modify the dimensions of an input array to a specified shape. More...
 
AFAPI array moddims (const array &in, const unsigned ndims, const dim_t *const dims)
 C++ Interface to modify the dimensions of an input array to a specified shape. More...
 
AFAPI array flat (const array &in)
 C++ Interface to flatten an array. More...
 
AFAPI array flip (const array &in, const unsigned dim)
 C++ Interface to flip an array. More...
 
AFAPI array lower (const array &in, bool is_unit_diag=false)
 C++ Interface to return the lower triangle array. More...
 
AFAPI array upper (const array &in, bool is_unit_diag=false)
 C++ Interface to return the upper triangle array. More...
 
AFAPI array select (const array &cond, const array &a, const array &b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI array select (const array &cond, const array &a, const double &b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI array select (const array &cond, const double &a, const array &b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI void replace (array &a, const array &cond, const array &b)
 C++ Interface to replace elements of an array with elements of another array. More...
 
AFAPI void replace (array &a, const array &cond, const double &b)
 C++ Interface to replace elements of an array with a scalar value. More...
 
AFAPI array pad (const array &in, const dim4 &beginPadding, const dim4 &endPadding, const borderType padFillType)
 C++ Interface to pad an array. More...
 
AFAPI void replace (array &a, const array &cond, const long long b)
 C++ Interface to replace elements of an array with a scalar value. More...
 
AFAPI void replace (array &a, const array &cond, const unsigned long long b)
 C++ Interface to replace elements of an array with a scalar value. More...
 
AFAPI array select (const array &cond, const array &a, const long long b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI array select (const array &cond, const array &a, const unsigned long long b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI array select (const array &cond, const long long a, const array &b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI array select (const array &cond, const unsigned long long a, const array &b)
 C++ Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_constant (af_array *arr, const double val, const unsigned ndims, const dim_t *const dims, const af_dtype type)
 C Interface to generate an array with elements set to a specified value. More...
 
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)
 C Interface to generate a complex array with elements set to a specified value. More...
 
AFAPI af_err af_constant_long (af_array *arr, const long long val, const unsigned ndims, const dim_t *const dims)
 C Interface to generate an array with elements set to a specified value. More...
 
AFAPI af_err af_constant_ulong (af_array *arr, const unsigned long long val, const unsigned ndims, const dim_t *const dims)
 C Interface to generate an array with elements set to a specified value. More...
 
AFAPI af_err af_identity (af_array *out, const unsigned ndims, const dim_t *const dims, const af_dtype type)
 C Interface to generate an identity array. More...
 
AFAPI af_err af_range (af_array *out, const unsigned ndims, const dim_t *const dims, const int seq_dim, const af_dtype type)
 C Interface to generate an array with [0, n-1] values along the seq_dim dimension and tiled across other dimensions of shape dim4. More...
 
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)
 C Interface to generate an array with [0, n-1] values modified to specified dimensions and tiling. More...
 
AFAPI af_err af_diag_create (af_array *out, const af_array in, const int num)
 C Interface to create a diagonal matrix from an extracted diagonal array. More...
 
AFAPI af_err af_diag_extract (af_array *out, const af_array in, const int num)
 C Interface to extract the diagonal from an array. More...
 
AFAPI af_err af_join (af_array *out, const int dim, const af_array first, const af_array second)
 C Interface to join 2 arrays along a dimension. More...
 
AFAPI af_err af_join_many (af_array *out, const int dim, const unsigned n_arrays, const af_array *inputs)
 C Interface to join many arrays along a dimension. More...
 
AFAPI af_err af_tile (af_array *out, const af_array in, const unsigned x, const unsigned y, const unsigned z, const unsigned w)
 C Interface to generate a tiled array. More...
 
AFAPI af_err af_reorder (af_array *out, const af_array in, const unsigned x, const unsigned y, const unsigned z, const unsigned w)
 C Interface to reorder an array. More...
 
AFAPI af_err af_shift (af_array *out, const af_array in, const int x, const int y, const int z, const int w)
 C Interface to shift an array. More...
 
AFAPI af_err af_moddims (af_array *out, const af_array in, const unsigned ndims, const dim_t *const dims)
 C Interface to modify the dimensions of an input array to a specified shape. More...
 
AFAPI af_err af_flat (af_array *out, const af_array in)
 C Interface to flatten an array. More...
 
AFAPI af_err af_flip (af_array *out, const af_array in, const unsigned dim)
 C Interface to flip an array. More...
 
AFAPI af_err af_lower (af_array *out, const af_array in, bool is_unit_diag)
 C Interface to return the lower triangle array. More...
 
AFAPI af_err af_upper (af_array *out, const af_array in, bool is_unit_diag)
 C Interface to return the upper triangle array. More...
 
AFAPI af_err af_select (af_array *out, const af_array cond, const af_array a, const af_array b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_select_scalar_r (af_array *out, const af_array cond, const af_array a, const double b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_select_scalar_l (af_array *out, const af_array cond, const double a, const af_array b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_replace (af_array a, const af_array cond, const af_array b)
 C Interface to replace elements of an array with elements of another array. More...
 
AFAPI af_err af_replace_scalar (af_array a, const af_array cond, const double b)
 C Interface to replace elements of an array with a scalar value. More...
 
AFAPI af_err af_pad (af_array *out, const af_array in, const unsigned begin_ndims, const dim_t *const begin_dims, const unsigned end_ndims, const dim_t *const end_dims, const af_border_type pad_fill_type)
 C Interface to pad an array. More...
 
AFAPI af_err af_replace_scalar_long (af_array a, const af_array cond, const long long b)
 C Interface to replace elements of an array with a scalar value. More...
 
AFAPI af_err af_replace_scalar_ulong (af_array a, const af_array cond, const unsigned long long b)
 C Interface to replace elements of an array with a scalar value. More...
 
AFAPI af_err af_select_scalar_r_long (af_array *out, const af_array cond, const af_array a, const long long b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_select_scalar_r_ulong (af_array *out, const af_array cond, const af_array a, const unsigned long long b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_select_scalar_l_long (af_array *out, const af_array cond, const long long a, const af_array b)
 C Interface to select elements based on a conditional array. More...
 
AFAPI af_err af_select_scalar_l_ulong (af_array *out, const af_array cond, const unsigned long long a, const af_array b)
 C Interface to select elements based on a conditional array. More...