secmlt.utils package#

Submodules#

secmlt.utils.tensor_utils module#

Basic utils for tensor handling.

secmlt.utils.tensor_utils.atleast_kd(x: Tensor, k: int) Tensor[source]#

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.