afe.ir.torch_utils

Helper functions using PyTorch API

Functions

convert_numpy_to_torch(→ torch.Tensor)
torch_tensor_to_scalar(→ Union[int, float])
numpy_tensor_to_scalar(→ Union[int, float])

Module Contents

afe.ir.torch_utils.convert_numpy_to_torch(x: numpy.ndarray, layout: str = 'NHWC') → torch.Tensor

afe.ir.torch_utils.torch_tensor_to_scalar(x: torch.Tensor) → int | float

afe.ir.torch_utils.numpy_tensor_to_scalar(x: numpy.ndarray) → int | float