//
// Created by xiangzhou on 11/9/23.
//

#pragma once
#include "C_Tensor.h"
#ifdef __cplusplus
extern "C" {
#endif

//----------------------------------------------------------------------------
//      Multinomial
//------------------------------------------------------------------------
struct MultinomialAttribute {
    // user
    const char* name_;
    int sample_size_;
    float seed_;
};
typedef struct MultinomialAttribute MultinomialAttribute;

//----------------------------------------------------------------------------
//      RandomUniformLike
//------------------------------------------------------------------------
struct RandomUniformLikeAttribute {
    // user
    const char* name_;
    float high_;
    float low_;
    float seed_;
};
typedef struct RandomUniformLikeAttribute RandomUniformLikeAttribute;

CREATE_DECL_RUN_FUN(Multinomial)
CREATE_DECL_RUN_FUN(RandomUniformLike)

#ifdef __cplusplus
}
#endif