histc.py

if __name__ == '__main__':
    import torch

    torch.set_default_device('cuda')

    t1 = torch.tensor([1, 2, 3, 3, 4, 5, 6, 7, 8], dtype=torch.float)
    res1 = torch.histc(t1, 5, 7, 7)
    print(res1)

    t2 = torch.tensor([7], dtype=torch.float)
    res2 = torch.histc(t2, 5, 7, 7)
    print(res2)
    
    t3 = torch.tensor([1, 2, 3, 3, 4, 5, 6, 7, 8], dtype=torch.float)
    res3 = torch.histc(t3, 5, 0, 0)
    print(res3)