parameters: (output_state, num_materials=4, *args, **kwargs)
return_statement: return make_one_shape_only(output_state) * np.digitize(output_state, bins)
docstring: Discretize outputs into bins, one for each material.
docstring_summary: Discretize outputs into bins, one for each material.
docstring_tokens: Discretize outputs into bins one for each material .
function: def discretize_material(output_state, num_materials=4, *args, **kwargs):
      """Discretize outputs into bins, one for each material."""
      bins = np.linspace(-1, 1, num=num_materials+1)
      return make_one_shape_only(output_state) * np.digitize(output_state, bins)
function_tokens: def discretize_material ( output_state , num_materials = 4 , * args , * * kwargs ) : bins = np . linspace ( - 1 , 1 , num = num_materials + 1 ) return make_one_shape_only ( output_state ) * np . digitize ( output_state , bins )