将指定位置设置为value值或者生成一个指定Shape的Tensor并填充为value。

输入x不可以被broadcast。

输入mask的元素只能是0或者1，且可以被broadcast。

struct FillParam {
    bool withMask = true;
    SVector<float> value;
    SVector<int64_t> outDim;
}

成员名称	描述
withMask	是否Masked Fill。
value	填充的元素，value是一个只含有一个元素的SVector。
outDim	withMask = false时，表示输出Tensor的Shape。

参数	维度	数据类型	格式	描述
x	[-1,…,-1] -1表示当前维度的大小没有约束。	float16/int32	ND	withMask = true时输入。
mask	[-1,…,-1] -1表示当前维度的大小没有约束。	int8/bool	ND	withMask = true时输入。

参数

维度

数据类型

格式

描述

x

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/int32

ND

withMask = true时输入。

mask

[-1,…,-1]

-1表示当前维度的大小没有约束。

int8/bool

ND

withMask = true时输入。

参数	维度	数据类型	格式	描述
output	[-1,…,-1] -1表示当前维度的大小没有约束。	float16/int32	ND	输出的Shape，withMask = true时和输入相同。 withMask = false时由outDim指定。

参数

维度

数据类型

格式

描述

output

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/int32

ND

输出的Shape，withMask = true时和输入相同。

withMask = false时由outDim指定。

FillOperation