Module: Tensorflow::Random

Defined in:
lib/tensorflow/ops/random.rb

Class Method Summary collapse

Class Method Details

.normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil) ⇒ Object



3
4
5
6
7
# File 'lib/tensorflow/ops/random.rb', line 3

def self.normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil)
  seed ||= ::Random.rand
  random = RawOps.random_standard_normal(shape, seed: seed, seed2: seed, dtype: dtype)
  mul = (random * stddev) + mean
end

.truncated_normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil) ⇒ Object



9
10
11
12
# File 'lib/tensorflow/ops/random.rb', line 9

def self.truncated_normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil)
  random = RawOps.truncated_normal(shape, seed: seed, seed2: seed, dtype: dtype)
  mul = (random * stddev) + mean
end

.uniform(shape, seed: 0, seed2: 0, dtype: :float, typeT: nil) ⇒ Object



14
15
16
# File 'lib/tensorflow/ops/random.rb', line 14

def self.uniform(shape, seed: 0, seed2: 0, dtype: :float, typeT: nil)
  RawOps.random_uniform(shape, seed: seed, seed2: seed2, dtype: dtype, typeT: typeT)
end