is_scalar
函数功能
根据tensor的shape校验是否是标量。
函数原型
def is_scalar(shape)
参数说明
参数 |
说明 |
|---|---|
shape |
需要校验的tensor的shape。 |
返回值说明
如果是标量，则返回True。否则，返回False。
约束说明
无
调用示例
from tbe.common.utils import para_check para_check.is_scalar((1,))
父主题： 算子参数校验