Method getSubTensor

Core/VirtualQuantumProcessor/SingleAmplitude/Tensor.cpp:102–149

}

void CPUComplexTensor::getSubTensor(size_t num, int value)
{
    if (num > m_rank)
    {
        QCERR("getSubTensor error");
        throw std::runtime_error("getSubTensor error");
    }

    auto size = 1ull << m_rank;
    auto sub = m_rank - num;
    qsize_t step = 1ull << sub;

    m_rank = this->m_rank - 1;
    auto new_size = 1ull << m_rank;
    auto new_tensor = (qcomplex_data_t *)calloc(new_size, sizeof(qcomplex_data_t));
    if (nullptr == new_tensor)
    {
        QCERR("calloc_fail");
        throw calloc_fail();
    }

    int threads = get_num_threads(m_rank);
    size_t j;
    size_t k = 0;
#pragma omp parallel for num_threads(threads) private(j,k)
    for (long long i = 0; i < size; i += step * 2)
    {
        k = i / (2 * step);
        for (j = i; j < i + step; j++)
        {
            if (0 == value)
            {
                new_tensor[j - k * step] = m_tensor[j];
            }
            else if (1 == value)
            {
                new_tensor[j - k * step] = m_tensor[j + step];
            }
            else
            {
                throw std::runtime_error("error");
            }
        }
    }

    free(m_tensor);
    m_tensor = new_tensor;
}

void CPUComplexTensor::dimDecrement(size_t num)
{