Method: TensorStream::MathGradients._broadcast_gradient_args

Defined in:
lib/tensor_stream/math_gradients.rb

._broadcast_gradient_args(input_a, input_b) ⇒ Object



192
193
194
195
# File 'lib/tensor_stream/math_gradients.rb', line 192

def self._broadcast_gradient_args(input_a, input_b)
  res = _op(:broadcast_gradient_args, input_a, input_b)
  [res[0], res[1]]
end