gm.data.Pad

gm.data.Pad#

class gemma.gm.data.Pad(*, key: typing.Annotated[typing.Any, <object object at 0x75a909cb7ae0>] | typing.Sequence[typing.Annotated[typing.Any, <object object at 0x75a909cb7ae0>]] | dict[typing.Annotated[typing.Any, <object object at 0x75a909cb7ae0>], typing.Annotated[typing.Any, <object object at 0x75a909cb7ae0>]], max_length: int, truncate: bool = False)[source]

基类:kauldron.data.transforms.base.ElementWiseTransform

在序列末尾添加零以达到最大长度。

max_length

序列的最大长度。

类型:

int

truncate

是否将序列截断到最大长度。如果为 False,则长度超过 max_length 的序列将引发错误。

类型:

bool

max_length: int
truncate: bool = False
map_element(
element: jaxtyping.Shaped[Array, 'length'] | jaxtyping.Shaped[ndarray, 'length'],
) jaxtyping.Shaped[Array, 'max_length'] | jaxtyping.Shaped[ndarray, 'max_length'][source]