Method: TensorStream::MathGradients._reshape_to_input

Defined in:
lib/tensor_stream/math_gradients.rb

._reshape_to_input(node, grad) ⇒ Object 



188
189
190
 
# File 'lib/tensor_stream/math_gradients.rb', line 188

def self._reshape_to_input(node, grad)
  ts.reshape(grad, ts.shape(node.inputs[0]))
end