Method: TensorFlow::Math.reduce_variance

Defined in:
lib/tensorflow/math.rb

.reduce_variance(input_tensor, axis: nil, keepdims: false) ⇒ Object



355
356
357
358
359
# File 'lib/tensorflow/math.rb', line 355

def reduce_variance(input_tensor, axis: nil, keepdims: false)
  means = reduce_mean(input_tensor, axis: axis, keepdims: true)
  squared_deviations = RawOps.square(x: input_tensor - means)
  reduce_mean(squared_deviations, axis: axis, keepdims: keepdims)
end