sum
该OP是对指定维度上的Tensor元素进行求和运算,并输出相应的计算结果。
参数:
x (Tensor)- 输入变量为多维Tensor,支持数据类型为float32,float64,int32,int64。
dtype (str , 可选)- 输出变量的数据类型。若参数为空,则输出变量的数据类型和输入变量相同,默认值为None。
name (str , 可选)- 具体用法请参见 ,一般无需设置,默认值为None。
返回:
Tensor
,在指定维度上进行求和运算的Tensor,数据类型和输入数据类型一致。
代码示例