https://github.com/rusty1s/pytorch_bincount
https://github.com/rusty1s/pytorch_bincount
Last synced: 9 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/rusty1s/pytorch_bincount
- Owner: rusty1s
- License: mit
- Created: 2018-06-05T07:04:38.000Z (about 8 years ago)
- Default Branch: master
- Last Pushed: 2018-10-03T09:05:42.000Z (almost 8 years ago)
- Last Synced: 2025-05-09T01:14:20.063Z (about 1 year ago)
- Language: Python
- Size: 12.7 KB
- Stars: 6
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
[pypi-image]: https://badge.fury.io/py/torch-bincount.svg
[pypi-url]: https://pypi.python.org/pypi/torch-bincount
[build-image]: https://travis-ci.org/rusty1s/pytorch_bincount.svg?branch=master
[build-url]: https://travis-ci.org/rusty1s/pytorch_bincount
[coverage-image]: https://codecov.io/gh/rusty1s/pytorch_bincount/branch/master/graph/badge.svg
[coverage-url]: https://codecov.io/github/rusty1s/pytorch_bincount?branch=master
# PyTorch BinCount
[![PyPI Version][pypi-image]][pypi-url]
[![Build Status][build-image]][build-url]
[![Code Coverage][coverage-image]][coverage-url]
--------------------------------------------------------------------------------
**PyTorch 0.4.1 now supports [`bincount`](https://pytorch.org/docs/stable/torch.html#torch.bincount) both for CPU and GPU.
Therefore, this package is no longer needed and will not be updated.**
--------------------------------------------------------------------------------
This package consists of a small extension library of a highly optimized `bincount` operation for the use in [PyTorch](http://pytorch.org/), which is missing in the main package.
The operation works on varying data types and is implemented both for CPU and GPU.
## Installation
Ensure that at least PyTorch 0.4.1 is installed and verify that `cuda/bin` and `cuda/include` are in your `$PATH` and `$CPATH` respectively, *e.g.*:
```
$ python -c "import torch; print(torch.__version__)"
>>> 0.4.1
$ echo $PATH
>>> /usr/local/cuda/bin:...
$ echo $CPATH
>>> /usr/local/cuda/include:...
```
Then run:
```
pip install torch-bincount
```
If you are running into any installation problems, please create an [issue](https://github.com/rusty1s/pytorch_bincount/issues).
Be sure to import `torch` first before using this package to resolve symbols the dynamic linker must see.
## Usage
```
torch_bincount.bincount(src, size=None) -> LongTensor
```
Counts the number of occurrences of each value in a non-negative tensor.
### Parameters
* **src** *(Tensor)* - The input tensor.
* **size** *(int, optional)* - The maximum number of bins for the output array. (default: `None`)
### Returns
* **out** *(LongTensor)* - The result of binning the input tensor.
### Example
```py
import torch
from torch_bincount import bincount
src = torch.tensor([2, 1, 1, 2, 4, 4, 2])
out = bincount(src)
```
```
print(out)
tensor([ 0, 2, 3, 0, 2])
```
## Running tests
```
python setup.py test
```