IsNaN

Versioned name: IsNaN-10

Category: Comparison

Short description: IsNaN returns the boolean mask of a given tensor which maps NaN to True.

Detailed description: IsNaN returns the boolean mask of the input tensor in which True corresponds to NaN and False to other values. The output tensor has the same shape as the input tensor. The i’th element of the output tensor is True if i’th element of the input tensor is NaN. Otherwise, it is False. For example, for the given input tensor [NaN, 2.1, 3.7, NaN, Inf] the output tensor is [True, False, False, True, False].

Attributes

IsNaN operation has no attributes.

Inputs:

  • 1: A tensor of type T_IN and arbitrary shape. Required.

Outputs:

  • 1: A boolean tensor of the same shape as the input tensor.

Types

  • T_IN: any supported floating-point type.

Example

<layer ... type="IsNaN">
    <input>
        <port id="0" precision="FP32">
            <dim>256</dim>
            <dim>56</dim>
        </port>
     </input>
    <output>
        <port id="1" precision="BOOL">
            <dim>256</dim>
            <dim>56</dim>
        </port>
    </output>
</layer>