(inp, inpr)
return sampling_module.prob_sample(inp, inpr)
input:
    batch_size * ncategory float32
    batch_size * npoints   float32
returns:
    batch_size * npoints   int32
input:
    batch_size * ncategory float32
    batch_size * npoints   float32
returns:
    batch_size * npoints   int32
input : batch_size * ncategory float32 batch_size * npoints float32 returns : batch_size * npoints int32
def prob_sample(inp, inpr):
    """
input:
    batch_size * ncategory float32
    batch_size * npoints   float32
returns:
    batch_size * npoints   int32
    """
    return sampling_module.prob_sample(inp, inpr)
def prob_sample ( inp , inpr ) : return sampling_module . prob_sample ( inp , inpr )