mars.tensor.iscomplex

mars.tensor.iscomplex(x, **kwargs)

返回一个布尔张量，如果输入元素是复数则为True。

测试的是输入是否具有非零的虚部，而不是输入类型是否为复数。

Parameters

x (array_like) – 输入张量。

Returns

out – 输出张量。

Return type

布尔值张量

另请参阅

isreal

iscomplexobj

如果 x 是复数类型或复数数组，则返回 True。

示例

>>> import mars.tensor as mt

>>> mt.iscomplex([1+1j, 1+0j, 4.5, 3, 2, 2j]).execute()
array([ True, False, False, False, False,  True])