Function SizeToAxis

dnn/src/cambricon/softmax/opr_impl.cpp:16–22

}

size_t SizeToAxis(const int axis, const size_t* dims) {
    size_t size = 1;
    for (int i = 0; i < axis; i++) {
        size *= dims[i];
    }
    return size;
}

size_t SizeOutAxis(const int axis, const size_t* dims, const int ndim) {
    size_t size = 1;