Method: TensorStream::MathGradients._broadcast_mul

Defined in:
lib/tensor_stream/math_gradients.rb

._broadcast_mul(vec, mat) ⇒ Object 



238
239
240
241
 
# File 'lib/tensor_stream/math_gradients.rb', line 238

def self._broadcast_mul(vec, mat)
  vec = ts.expand_dims(vec, -1)
  vec * mat
end