Module: Tensorflow::Random
- Defined in:
- lib/tensorflow/ops/random.rb
Class Method Summary collapse
- .normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil) ⇒ Object
- .truncated_normal(shape, mean: 0.0, stddev: 1.0, dtype: :float, seed: nil, name: nil) ⇒ Object
- .uniform(shape, seed: 0, seed2: 0, dtype: :float, typeT: nil) ⇒ Object
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 |