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 |