C++ (Cpp) out_slice示例

示例#1

文件： subview_cube_each_meat.hpp 项目： KaimingOuyang/HPC-K-Means

inline
Cube<eT>
subview_cube_each1_aux::operator_div
  (
  const Base<eT,T1>&            X,
  const subview_cube_each1<eT>& Y
  )
  {
  arma_extra_debug_sigprint();
  
  const Cube<eT>& p = Y.P;
  
  const uword p_n_rows   = p.n_rows;
  const uword p_n_cols   = p.n_cols;
  const uword p_n_slices = p.n_slices;
  
  Cube<eT> out(p_n_rows, p_n_cols, p_n_slices);
  
  const unwrap<T1>   tmp(X.get_ref());
  const Mat<eT>& A = tmp.M;
  
  Y.check_size(A);
  
  for(uword i=0; i < p_n_slices; ++i)
    {
          Mat<eT> out_slice(              out.slice_memptr(i),  p_n_rows, p_n_cols, false, true);
    const Mat<eT>   p_slice(const_cast<eT*>(p.slice_memptr(i)), p_n_rows, p_n_cols, false, true);
    
    out_slice = A / p_slice;
    }
  
  return out;
  }

示例#2

文件： subview_cube_each_meat.hpp 项目： RcppCore/RcppArmadillo

inline
Cube<eT>
subview_cube_each1_aux::operator_times
  (
  const Base<eT,T1>&            X,
  const subview_cube_each1<eT>& Y
  )
  {
  arma_extra_debug_sigprint();
  
  const unwrap<T1>   tmp(X.get_ref());
  const Mat<eT>& M = tmp.M;
  
  const Cube<eT>& C = Y.P;
  
  Cube<eT> out(M.n_rows, C.n_cols, C.n_slices);
  
  for(uword i=0; i < C.n_slices; ++i)
    {
          Mat<eT> out_slice(              out.slice_memptr(i),  M.n_rows, C.n_cols, false, true);
    const Mat<eT>   C_slice(const_cast<eT*>(C.slice_memptr(i)), C.n_rows, C.n_cols, false, true);
    
    out_slice = M * C_slice;
    }
  
  return out;
  }

示例#3

文件： subview_cube_each_meat.hpp 项目： KaimingOuyang/HPC-K-Means

inline
Cube<eT>
subview_cube_each2_aux::operator_div
  (
  const Base<eT,T1>&               X,
  const subview_cube_each2<eT,TB>& Y
  )
  {
  arma_extra_debug_sigprint();
  
  const Cube<eT>& p = Y.P;
  
  const uword p_n_rows   = p.n_rows;
  const uword p_n_cols   = p.n_cols;
  const uword p_n_slices = p.n_slices;
  
  Cube<eT> out = p;
  
  const unwrap<T1>   tmp(X.get_ref());
  const Mat<eT>& A = tmp.M;
  
  const unwrap<TB> U(Y.base_indices.get_ref());
  
  Y.check_size(A);
  Y.check_indices(U.M);
  
  const uword* indices_mem = U.M.memptr();
  const uword  N           = U.M.n_elem;
  
  for(uword i=0; i < N; ++i)
    {
    const uword slice = indices_mem[i];
    
    arma_debug_check( (slice >= p_n_slices), "each_slice(): index out of bounds" );
    
          Mat<eT> out_slice(              out.slice_memptr(slice),  p_n_rows, p_n_cols, false, true);
    const Mat<eT>   p_slice(const_cast<eT*>(p.slice_memptr(slice)), p_n_rows, p_n_cols, false, true);
    
    out_slice = A / p_slice;
    }
  
  return out;
  }