IsFloatingType

函数功能

判断输入的数据类型是否为浮点类型，包括Float64（即Double）、Float32（即Float）、BFloat16、Float16。

函数原型

bool IsFloatingType(const DataType dtype)

参数说明

返回值说明

若为浮点类型返回true，否则返回false。

约束说明

无

调用示例

1
2
3
4
5
6

// 判断dtype不为浮点数类型时，返回
void Func(const DataType dtype) {
    if (!IsFloatingType(dtype)) {
        return;
    }
}