Class: Torch::Distributions::Distribution

Inherits:
Object
  • Object
show all
Defined in:
lib/torch/distributions/distribution.rb

Direct Known Subclasses

ExponentialFamily

Instance Method Summary collapse

Constructor Details

#initialize(batch_shape: [], event_shape: [], validate_args: nil) ⇒ Distribution

Returns a new instance of Distribution.



4
5
6
7
8
9
10
11
12
13
14
 
# File 'lib/torch/distributions/distribution.rb', line 4

def initialize(batch_shape: [], event_shape: [], validate_args: nil)
  @batch_shape = batch_shape
  @event_shape = event_shape
  if !validate_args.nil?
    @validate_args = validate_args
  end
  if @validate_args
    raise NotImplementedYet
  end
  super()
end