NonZero - 9 vs 13

Files changed (1) hide show
  1. NonZero9 → NonZero13 +1 -0
NonZero9 → NonZero13 RENAMED
@@ -1 +1 @@
1
1
  Returns the indices of the elements that are non-zero
2
2
  (in row-major order - by dimension).
3
3
  NonZero behaves similar to numpy.nonzero:
4
4
  https://docs.scipy.org/doc/numpy/reference/generated/numpy.nonzero.html,
5
5
  but for scalar input, NonZero produces output shape (0, N) instead of (1, N), which is different from Numpy's behavior.
6
6
  **Inputs**
7
7
  * **X** (heterogeneous) - **T**:
8
8
  input
9
9
  **Outputs**
10
10
  * **Y** (heterogeneous) - **tensor(int64)**:
11
11
  output
12
12
  **Type Constraints**
13
13
  * **T** in (
14
+ tensor(bfloat16),
14
15
  tensor(bool),
15
16
  tensor(complex128),
16
17
  tensor(complex64),
17
18
  tensor(double),
18
19
  tensor(float),
19
20
  tensor(float16),
20
21
  tensor(int16),
21
22
  tensor(int32),
22
23
  tensor(int64),
23
24
  tensor(int8),
24
25
  tensor(string),
25
26
  tensor(uint16),
26
27
  tensor(uint32),
27
28
  tensor(uint64),
28
29
  tensor(uint8)
29
30
  ):
30
31
  Constrain to all tensor types.