Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Attributes | List of all members
torch.distributions.geometric.Geometric Class Reference
Inheritance diagram for torch.distributions.geometric.Geometric:
torch.distributions.distribution.Distribution

Public Member Functions

def __init__ (self, probs=None, logits=None, validate_args=None)
 
def expand (self, batch_shape, _instance=None)
 
def mean (self)
 
def variance (self)
 
def logits (self)
 
def probs (self)
 
def sample (self, sample_shape=torch.Size())
 
def log_prob (self, value)
 
def entropy (self)
 
- Public Member Functions inherited from torch.distributions.distribution.Distribution
def __init__ (self, batch_shape=torch.Size(), event_shape=torch.Size(), validate_args=None)
 
def expand (self, batch_shape, _instance=None)
 
def batch_shape (self)
 
def event_shape (self)
 
def arg_constraints (self)
 
def support (self)
 
def mean (self)
 
def variance (self)
 
def stddev (self)
 
def sample (self, sample_shape=torch.Size())
 
def rsample (self, sample_shape=torch.Size())
 
def sample_n (self, n)
 
def log_prob (self, value)
 
def cdf (self, value)
 
def icdf (self, value)
 
def enumerate_support (self, expand=True)
 
def entropy (self)
 
def perplexity (self)
 
def __repr__ (self)
 

Static Public Attributes

 arg_constraints
 
 support
 
- Static Public Attributes inherited from torch.distributions.distribution.Distribution
 has_rsample
 
 has_enumerate_support
 
 support
 
 arg_constraints
 

Additional Inherited Members

- Static Public Member Functions inherited from torch.distributions.distribution.Distribution
def set_default_validate_args (value)
 

Detailed Description

Definition at line 10 of file geometric.py.


The documentation for this class was generated from the following file: