`pipert.utils.structures.layers`¶

Module Contents¶

Functions¶

cat(tensors, dim=0)

Efficient version of torch.cat that avoids a copy if there is only a single element in a list

pipert.utils.structures.layers.cat(tensors, dim=0)[source]¶: Efficient version of torch.cat that avoids a copy if there is only a single element in a list