FM / pybind11

/usr/local/lib64/python3.6/site-packages/torch/include/pybind11/

/usr/local/lib64/python3.6/site-packages/torch/include/pybind11

Name	Size	Mode	Actions
detail/	-	0755	rm
attr.h	21412	0644	edit dl rm
buffer_info.h	6118	0644	edit dl rm
cast.h	95557	0644	edit dl rm
chrono.h	8185	0644	edit dl rm
common.h	120	0644	edit dl rm
complex.h	2037	0644	edit dl rm
eigen.h	29086	0644	edit dl rm
embed.h	7843	0644	edit dl rm
eval.h	5079	0644	edit dl rm
functional.h	3709	0644	edit dl rm
iostream.h	6084	0644	edit dl rm
numpy.h	69310	0644	edit dl rm
operators.h	9085	0644	edit dl rm
options.h	2049	0644	edit dl rm
pybind11.h	111558	0644	edit dl rm
pytypes.h	66118	0644	edit dl rm
stl.h	14136	0644	edit dl rm
stl_bind.h	23912	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/pybind11/complex.h (2037B)

/*
    pybind11/complex.h: Complex number support

All rights reserved. Use of this source code is governed by a
    BSD-style license that can be found in the LICENSE file.
*/

#pragma once

#include "pybind11.h"
#include <complex>

/// glibc defines I as a macro which breaks things, e.g., boost template names
#ifdef I
#  undef I
#endif

PYBIND11_NAMESPACE_BEGIN(PYBIND11_NAMESPACE)

template <typename T> struct format_descriptor<std::complex<T>, detail::enable_if_t<std::is_floating_point<T>::value>> {
    static constexpr const char c = format_descriptor<T>::c;
    static constexpr const char value[3] = { 'Z', c, '\0' };
    static std::string format() { return std::string(value); }
};

#ifndef PYBIND11_CPP17

template <typename T> constexpr const char format_descriptor<
    std::complex<T>, detail::enable_if_t<std::is_floating_point<T>::value>>::value[3];

#endif

PYBIND11_NAMESPACE_BEGIN(detail)

template <typename T> struct is_fmt_numeric<std::complex<T>, detail::enable_if_t<std::is_floating_point<T>::value>> {
    static constexpr bool value = true;
    static constexpr int index = is_fmt_numeric<T>::index + 3;
};

template <typename T> class type_caster<std::complex<T>> {
public:
    bool load(handle src, bool convert) {
        if (!src)
            return false;
        if (!convert && !PyComplex_Check(src.ptr()))
            return false;
        Py_complex result = PyComplex_AsCComplex(src.ptr());
        if (result.real == -1.0 && PyErr_Occurred()) {
            PyErr_Clear();
            return false;
        }
        value = std::complex<T>((T) result.real, (T) result.imag);
        return true;
    }

static handle cast(const std::complex<T> &src, return_value_policy /* policy */, handle /* parent */) {
        return PyComplex_FromDoubles((double) src.real(), (double) src.imag());
    }

PYBIND11_TYPE_CASTER(std::complex<T>, _("complex"));
};
PYBIND11_NAMESPACE_END(detail)
PYBIND11_NAMESPACE_END(PYBIND11_NAMESPACE)

cmd: