Input对象初始化。
def __init__(self, name, shape, datatype)
参数名 |
参数类型 |
输入/输出 |
说明 |
|---|---|---|---|
name |
str |
输入 |
输入名称。 |
shape |
array |
输入 |
输入形状。 只支持[1,n]，n为tokenid数量。 |
datatype |
str |
输入 |
输入名称数据类型。 只支持UINT32。 |
无