assert를 이용해 shape/차원 확인하기

목차

assert를 이용해 shape/차원 확인하기

• 넘파이를 이용해 코드를 작성하거나, 모델을 작성하다 보면 shape이나 차원이 헷갈리는 경우가 다수 존재합니다.
• 이럴 경우 항상 주저하지 말고 assert를 이용하여 디버깅하는 습관을 들여야 합니다.

예시 코드

• numpy

  • import numpy
    
    def process_array(array):
        # 입력 배열이 2차원이며, 크기가 3x4인지 확인
        assert array.ndim == 2, "배열은 2차원이어야 합니다."
        assert array.shape == (3, 4), "배열의 크기는 3x4여야 합니다."

• pytorch

  • import torch
    
    def process_tensor(tensor):
        # 입력 텐서가 2차원이며, 크기가 3x4인지 확인
        assert tensor.ndimension() == 2, "텐서는 2차원이어야 합니다."
        assert tensor.shape == (3, 4), "텐서의 크기는 3x4여야 합니다."