render_array_sharding

render_array_sharding#

treescope.render_array_sharding(array: ArrayInRegistry, rows: Sequence[int | AxisName] = (), columns: Sequence[int | AxisName] = ()) → figures_impl.TreescopeFigure[source]#

Renders the sharding of an array.

Parameters:

  • array – The array whose sharding we should render.

  • rows – Optional explicit ordering of axes for the visualization rows.

  • columns – Optional explicit ordering of axes for the visualization columns.

Returns:

A rendering of that array’s sharding.

Contents