fmelib.ComputeParam Namespace Reference

Functions
Tuple[List[float], List[float]]	_channelwiseMinMax (Tensors tensors, int channel)
np.ndarray	getActivationMax (np.ndarray tensor)

Variables
	Tensors = List[np.ndarray]

Function Documentation

_channelwiseMinMax()

Tuple[List[float], List[float]] fmelib.ComputeParam._channelwiseMinMax ( Tensors  tensors, int  channel  )

protected

Compute channel-wise min and max for the tensor.
:param tensors: a list of numpy array (each is a tensor)
:param channel: number of channels
:return: lists of min and max for each channel

Definition at line 25 of file ComputeParam.py.

def _channelwiseMinMax(tensors: Tensors, channel: int) -> Tuple[List[float], List[float]]:
    """
    Compute channel-wise min and max for the tensor.
    :param tensors: a list of numpy array (each is a tensor)
    :param channel: number of channels
    :return: lists of min and max for each channel
    """
    channel_wise_min = []
    channel_wise_max = []
    for c in range(channel):
        min_act = min(tensors, key=lambda activation: np.min(activation[:, :, :, c]))
        max_act = max(tensors, key=lambda activation: np.max(activation[:, :, :, c]))
        channel_wise_min.append(float(np.min(min_act[:, :, :, c])))
        channel_wise_max.append(float(np.max(max_act[:, :, :, c])))
    return channel_wise_min, channel_wise_max
 
 

getActivationMax()

np.ndarray fmelib.ComputeParam.getActivationMax

(

np.ndarray

tensor

)

Get max values of activation.
:param tensors: a list of numpy array (each is a tensor)
:return: 1D array of max absolute values for each channel

Definition at line 42 of file ComputeParam.py.

def getActivationMax(tensor: np.ndarray) -> np.ndarray:
    """
    Get max values of activation.
    :param tensors: a list of numpy array (each is a tensor)
    :return: 1D array of max absolute values for each channel
    """
    # max along with last dimension
    return np.abs(tensor.reshape(-1, tensor.shape[-1])).max(axis=0)

Variable Documentation

Tensors

fmelib.ComputeParam.Tensors = List[np.ndarray]

Definition at line 22 of file ComputeParam.py.