afe.ir.torch_utils
==================

.. py:module:: afe.ir.torch_utils

.. autoapi-nested-parse::

   Helper functions using PyTorch API



Functions
---------

.. autoapisummary::

   afe.ir.torch_utils.convert_numpy_to_torch
   afe.ir.torch_utils.torch_tensor_to_scalar
   afe.ir.torch_utils.numpy_tensor_to_scalar


Module Contents
---------------

.. py:function:: convert_numpy_to_torch(x: numpy.ndarray, layout: str = 'NHWC') -> torch.Tensor

.. py:function:: torch_tensor_to_scalar(x: torch.Tensor) -> Union[int, float]

.. py:function:: numpy_tensor_to_scalar(x: numpy.ndarray) -> Union[int, float]