-
Notifications
You must be signed in to change notification settings - Fork 52
Open
Labels
API extensionAdds new functions or objects to the API.Adds new functions or objects to the API.Needs DiscussionNeeds further discussion.Needs further discussion.RFCRequest for comments. Feature requests and proposed changes.Request for comments. Feature requests and proposed changes.
Description
It seems quite widely adopted:
- https://numpy.org/doc/stable/reference/generated/numpy.bincount.html
- https://pytorch.org/docs/stable/generated/torch.bincount.html
- https://docs.cupy.dev/en/stable/reference/generated/cupy.bincount.html
- https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.bincount.html
- https://docs.dask.org/en/stable/generated/dask.array.bincount.html
Although I have not checked how compatible they are.
JAX in particular requires an extra length
argument to statically define the length of the output array to make JIT possible and dask uses (nan,)
shape to represent a data-dependent output shapes.
Metadata
Metadata
Assignees
Labels
API extensionAdds new functions or objects to the API.Adds new functions or objects to the API.Needs DiscussionNeeds further discussion.Needs further discussion.RFCRequest for comments. Feature requests and proposed changes.Request for comments. Feature requests and proposed changes.