Computes the cumulative sum of a tensor along a specified axis.

tfb_cumsum(axis = -1, validate_args = FALSE, name = "cumsum")

Arguments

axis

int indicating the axis along which to compute the cumulative sum. Note that positive (and zero) values are not supported

validate_args

Logical, default FALSE. Whether to validate input with asserts. If validate_args is FALSE, and the inputs are invalid, correct behavior is not guaranteed.

name

name prefixed to Ops created by this class.

Value

a bijector instance.

See also