prokbert.general_utils.truncate_zero_columns

prokbert.general_utils.truncate_zero_columns(arr: ndarray) → ndarray

Truncate all trailing columns composed entirely of zeros in a given 2D numpy array.

Parameters

arr (np.ndarray) – Input 2D numpy array.

Returns

A new array with trailing zero columns removed.

Return type

np.ndarray