def set_summarize_num_elements(n: int):
  """Sets number of tensor elements to print in `Assert`s."""
  global _summarize_num_elements
  _summarize_num_elements = n
Sets number of tensor elements to print in `Assert`s.
language/canine/tensor_contracts.py