o i3@sddlmZmZmZmZddlmZddlmZm Z m Z ee e ee fZ ee ee dffZededZded eeeffd d Zded ee e ffd d Zded ee e ffddZddZdS))SequenceTupleTypeVarUnion)Model)ArrayXdFloatsXdIntsXd.ArrayTXd)boundindexreturncCtdtd|idS)a5Index into input arrays, and return the subarrays. index: A valid numpy-style index. Multi-dimensional indexing can be performed by passing in a tuple, and slicing can be performed using the slice object. For instance, X[:, :-1] would be (slice(None, None), slice(None, -1)). z array-getitemr attrsrforwardr rN/home/ubuntu/.local/lib/python3.10/site-packages/thinc/layers/array_getitem.py array_getitem srcCr)}Index into input arrays, and return the subarrays. This delegates to `array_getitem`, but allows type declarations. zfloats-getitemr rrrrrrfloats_getitemrcCr)rz ints-getitemr rrrrrr ints_getitemrrcsLjd|j|jfdd}t|dkr||fS|}||fS)Nr csjjd}||<|S)N)dtype)opsalloc)dYdXrr modelshaperrbackprop_get_column+sz$forward..backprop_get_columnr)rr#rlen)r"Xis_trainr$Yrr!rr&s  rN)typingrrrrr"rtypesrr r intslice AxisIndexIndexr rrrrrrrrs