Create a layout from a tensor of sequence lengths.
(seqlen: list)
| 29 | |
| 30 | @staticmethod |
| 31 | def layout_from_seqlen(seqlen: list) -> List[slice]: |
| 32 | """ |
| 33 | Create a layout from a tensor of sequence lengths. |
| 34 | """ |
| 35 | layout = [] |
| 36 | start = 0 |
| 37 | for l in seqlen: |
| 38 | layout.append(slice(start, start + l)) |
| 39 | start += l |
| 40 | return layout |
| 41 | |
| 42 | @staticmethod |
| 43 | def from_tensor_list(tensor_list: List[torch.Tensor]) -> 'VarLenTensor': |
nothing calls this directly
no outgoing calls
no test coverage detected