Method: TensorStream::MathGradients._op_supports_broadcast?

Defined in:
lib/tensor_stream/math_gradients.rb

._op_supports_broadcast?(node) ⇒ Boolean

Returns:

  • (Boolean) 


216
217
218
219
 
# File 'lib/tensor_stream/math_gradients.rb', line 216

def self._op_supports_broadcast?(node)
  return true if %i[add sub div mul pow].include?(node.operation)
  false
end