secmlt.utils package

Submodules

secmlt.utils.tensor_utils module

Basic utils for tensor handling.

secmlt.utils.tensor_utils.atleast_kd(x: torch.Tensor, k: int) → torch.Tensor

Add dimensions to the tensor x until it reaches k dimensions.

Parameters:

• x (torch.Tensor) – Input tensor.

• k (int) – Number of desired dimensions.

Returns:

The input tensor x but with k dimensions.

Return type:

torch.Tensor

Module contents

Utilities for the use of the library.