Class TensorShape

dnn/include/megdnn/basic_types.h:86–119

* \brief Describing the tensor shape. * * Uninitialized shape: ndim == 0; total_nr_elems() is also defined to be 0 * * Empty shape: ndim > 0 && shape[i] == 0 for 0 <= i < ndim; it is always * considered non-contiguous. */

 * considered non-contiguous.
 */
struct TensorShape {
    static MEGDNN_CONSTEXPR size_t MAX_NDIM = MEGDNN_MAX_NDIM;

#if MEGDNN_CC_HOST
    size_t shape[MAX_NDIM], ndim = 0;
#else
    size_t shape[MAX_NDIM], ndim;
#endif

#if MEGDNN_CC_HOST
    TensorShape() = default;
    TensorShape(const TensorShape& rhs) = default;
    MGE_WIN_DECLSPEC_FUC TensorShape& operator=(const TensorShape& rhs) = default;
    MGE_WIN_DECLSPEC_FUC TensorShape(const SmallVector<size_t>& init_shape);
    MGE_WIN_DECLSPEC_FUC TensorShape(std::initializer_list<size_t> init_shape);
    MGE_WIN_DECLSPEC_FUC std::string to_string() const;
#endif

    //! total number of elements
    MGE_WIN_DECLSPEC_FUC size_t total_nr_elems() const;

    //! check whether two shapes are equal
    MGE_WIN_DECLSPEC_FUC bool eq_shape(const TensorShape& rhs) const;

    //! check whether the shape can be treated as a scalar
    bool is_scalar() const { return ndim == 1 && shape[0] == 1; }

    //! check whether ndim != 0 and at least one shape is 0
    MGE_WIN_DECLSPEC_FUC bool is_empty() const;

    //! access single element, without boundary check
    size_t& operator[](size_t i) { return shape[i]; }
    size_t operator[](size_t i) const { return shape[i]; }
};

class Handle;
/**