(x)
| 49 | return (x[:, 0] | x[:, 1] << 8 | x[:, 2] << 16 | x[:, 3] << 24).unsqueeze(1) |
| 50 | |
| 51 | def to_uint16(x): |
| 52 | x = x.view(torch.uint8).to(torch.int32) |
| 53 | return (x[:, 0] | x[:, 1] << 8).unsqueeze(1) |
| 54 | |
| 55 | def split_block_dims(blocks, *args): |
| 56 | n_max = blocks.shape[1] |
no test coverage detected