scalar2tensor_one

函数功能

将标量shape（shape=[]）转换为[1]。

函数原型

def scalar2tensor_one(shape)

参数说明

返回值说明

如果输入是标量的shape，返回[1]；否则，直接返回输入。

约束说明

无

调用示例

from tbe.common.utils import shape_util 
shape1 = []
shape_util.scalar2tensor_one(shape1) 

返回shape为[1]。