Class ndarray

Source/Falcor/Utils/Scripting/ndarray.h:366–460

template<typename... Args>
class ndarray
{
public:
    using Info = detail::ndarray_info<Args...>;
    using Scalar = typename Info::scalar_type;

    ndarray() = default;

    explicit ndarray(detail::ndarray_handle* handle) : m_handle(handle)
    {
        if (handle)
            m_dltensor = *detail::ndarray_inc_ref(handle);
    }

    ndarray(
        void* value,
        size_t ndim,
        const size_t* shape,
        handle owner = pybind11::handle(),
        const int64_t* strides = nullptr,
        dlpack::dtype dtype = pybind11::dtype<Scalar>(),
        int32_t device_type = device::cpu::value,
        int32_t device_id = 0
    )
    {
        m_handle = detail::ndarray_create(value, ndim, shape, owner.ptr(), strides, &dtype, device_type, device_id);
        m_dltensor = *detail::ndarray_inc_ref(m_handle);
    }

    ~ndarray() { detail::ndarray_dec_ref(m_handle); }

    ndarray(const ndarray& t) : m_handle(t.m_handle), m_dltensor(t.m_dltensor) { detail::ndarray_inc_ref(m_handle); }

    ndarray(ndarray&& t) noexcept : m_handle(t.m_handle), m_dltensor(t.m_dltensor)
    {
        t.m_handle = nullptr;
        t.m_dltensor = dlpack::dltensor();
    }

    ndarray& operator=(ndarray&& t) noexcept
    {
        detail::ndarray_dec_ref(m_handle);
        m_handle = t.m_handle;
        m_dltensor = t.m_dltensor;
        t.m_handle = nullptr;
        t.m_dltensor = dlpack::dltensor();
        return *this;
    }

    ndarray& operator=(const ndarray& t)
    {
        detail::ndarray_inc_ref(t.m_handle);
        detail::ndarray_dec_ref(m_handle);
        m_handle = t.m_handle;
        m_dltensor = t.m_dltensor;
        return *this;
    }