Function modDims

src/backend/common/moddims.cpp:64–82

template<typename T>
Array<T> modDims(const Array<T> &in, const af::dim4 &newDims) {
    if (in.isLinear() == false) {
        // Nonlinear array's shape cannot be modified. Copy the data and modify
        // the shape of the array
        Array<T> out = copyArray<T>(in);
        out.setDataDims(newDims);
        return out;
    } else if (in.isReady()) {
        /// If the array is a buffer, modify the dimension and return
        auto out = in;
        out.setDataDims(newDims);
        return out;
    } else {
        /// If the array is a node and not linear and not a buffer, then create
        /// a moddims node
        auto out = moddimOp<T>(in, newDims);
        return out;
    }
}

template<typename T>
detail::Array<T> flat(const detail::Array<T> &in) {