SliceLike

class penzai.core.syntactic_sugar.SliceLike

Bases: object

Builds a slice when sliced (e.g. pz.slice[1:3] == slice(1, 3, None)).

An instance of this class is exposed as pz.slice, so that you can easily slice named array axes using syntax like arr[{"name": pz.slice[1:3]}] instead of arr[{"name": slice(1, 3, None)}].

Methods

__init__()