コード例 #1
0
ファイル: compare.py プロジェクト: celidos/TensorRT_study
 def compute_mean(buffer):
     if misc.is_empty_shape(buffer.shape):
         return 0
     return np.mean(buffer)
コード例 #2
0
ファイル: compare.py プロジェクト: celidos/TensorRT_study
 def compute_argmin(buffer):
     if misc.is_empty_shape(buffer.shape):
         return 0
     return np.unravel_index(np.argmin(buffer), buffer.shape)