def __init__(self, name, shape, datatype)
函数功能
Input对象初始化。
函数原型
def __init__(self, name, shape, datatype)
参数说明
参数名 |
参数类型 |
输入/输出 |
说明 |
---|---|---|---|
name |
str |
输入 |
输入名称,长度范围[1, 1000],name输入为空时,默认设为input0。 |
shape |
array |
输入 |
输入形状。 只支持[1,n],n为tokenid数量,且n<=16000。 |
datatype |
str |
输入 |
输入名称数据类型。 只支持UINT32。 |
返回值
无
父主题: class Input