CumProd#

Domain: ai.onnx
Since version: 26

Performs cumulative product of the input elements along the given axis. By default, it will do the product inclusively meaning the first element is copied as is. Through an exclusive attribute, this behavior can change to exclude the first element. It can also perform product in the opposite direction of the axis. For that, set reverse attribute to 1.

Example:

input_x = [1, 2, 3]
axis=0
output = [1, 2, 6]
exclusive=1
output = [1, 1, 2]
exclusive=0
reverse=1
output = [6, 6, 3]
exclusive=1
reverse=1
output = [6, 3, 1]

Inputs

x (T): An input tensor that is to be processed.
axis (T2): A 0-D tensor. Must be in the range [-rank(x), rank(x)-1]. Negative value means counting dimensions from the back.

Outputs

y (T): Output tensor of the same type as ‘x’ with cumulative products of the x’s elements

Attributes

exclusive (int): If set to 1 will return exclusive product in which the top element is not included. In other terms, if set to 1, the j-th output element would be the product of the first (j-1) elements. Otherwise, it would be the product of the first j elements.
reverse (int): If set to 1 will perform the products in reverse direction.

Type Constraints

T: Constrain input and output types to numeric tensors. Allowed types: tensor(bfloat16), tensor(double), tensor(float), tensor(float16), tensor(int32), tensor(int64), tensor(uint32), tensor(uint64).
T2: axis tensor can be int32 or int64 only Allowed types: tensor(int32), tensor(int64).

Examples#

test_cumprod_1d

Node:
  CumProd(x, axis) -> (y)

Inputs:
  x: shape=(3,), dtype=float64
    [1., 2., 3.]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(3,), dtype=float64
    [1., 2., 6.]

test_cumprod_1d_exclusive

Node:
  CumProd(x, axis) -> (y)
  Attributes:
    exclusive = 1

Inputs:
  x: shape=(3,), dtype=float64
    [1., 2., 3.]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(3,), dtype=float64
    [1., 1., 2.]

test_cumprod_1d_int32_exclusive

Node:
  CumProd(x, axis) -> (y)
  Attributes:
    exclusive = 1

Inputs:
  x: shape=(5,), dtype=int32
    [1, 2, 3, 4, 5]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(5,), dtype=int32
    [ 1,  1,  2,  6, 24]

test_cumprod_1d_reverse

Node:
  CumProd(x, axis) -> (y)
  Attributes:
    reverse = 1

Inputs:
  x: shape=(3,), dtype=float64
    [1., 2., 3.]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(3,), dtype=float64
    [6., 6., 3.]

test_cumprod_1d_reverse_exclusive

Node:
  CumProd(x, axis) -> (y)
  Attributes:
    exclusive = 1
    reverse = 1

Inputs:
  x: shape=(3,), dtype=float64
    [1., 2., 3.]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(3,), dtype=float64
    [6., 3., 1.]

test_cumprod_2d_axis_0

Node:
  CumProd(x, axis) -> (y)

Inputs:
  x: shape=(2, 3), dtype=float64
    [[1., 2., 3.],
     [4., 5., 6.]]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(2, 3), dtype=float64
    [[ 1.,  2.,  3.],
     [ 4., 10., 18.]]

test_cumprod_2d_axis_1

Node:
  CumProd(x, axis) -> (y)

Inputs:
  x: shape=(2, 3), dtype=float64
    [[1., 2., 3.],
     [4., 5., 6.]]
  axis: shape=(), dtype=int32
    1

Outputs:
  y: shape=(2, 3), dtype=float64
    [[  1.,   2.,   6.],
     [  4.,  20., 120.]]

test_cumprod_2d_int32

Node:
  CumProd(x, axis) -> (y)

Inputs:
  x: shape=(2, 3), dtype=int32
    [[1, 2, 3],
     [4, 5, 6]]
  axis: shape=(), dtype=int32
    0

Outputs:
  y: shape=(2, 3), dtype=int32
    [[ 1,  2,  3],
     [ 4, 10, 18]]

test_cumprod_2d_negative_axis

Node:
  CumProd(x, axis) -> (y)

Inputs:
  x: shape=(2, 3), dtype=float64
    [[1., 2., 3.],
     [4., 5., 6.]]
  axis: shape=(), dtype=int64
    -1

Outputs:
  y: shape=(2, 3), dtype=float64
    [[  1.,   2.,   6.],
     [  4.,  20., 120.]]