is there an equivalent of TensorFlow tf.segment_sum , or tf.unsorted_segment_sum ? I see there was conversation about adding this in PyTorch: Pytorch equivalent to tf.unsorted_segment_sum - PyTorch Forums

Actually that was overkill, just do: segment_sum = pt.zeros(n_segments+1)[segment_ids].inc(data) # debug eval segment_sum.eval() # array([13., 2., 7., 4.])

PyTensor version of TensorFlow segment_sum

Questions v5

ricardoV94 October 17, 2024, 9:42am 5

Because our documentation of the variable methods is not great. inc method is equivalent to the pt.inc_subtensor function which you should be able to find in the docs

1 Like

Topic		Replies	Views
How to vectorize indexing of variable for pm.math.sum? Questions	4	534	April 13, 2020
Using PyTensor.scan to loop over multidimensional array v5	13	1420	June 1, 2023
Iterating over Tensors - pytensor.scan v5 pytensor	8	750	June 7, 2023
How to use the "until" function required for conditional ending of scan in PyTensor version agnostic pytensor	1	410	April 26, 2023
How to write numerical summation from zero to inf by pytensor or pymc math modeling	1	124	March 12, 2024

PyTensor version of TensorFlow segment_sum

Related topics