브로드캐스트

rokky·2023년 11월 2일
0

딥러닝

목록 보기
15/18

넘파이(numpy)

  • 딥러닝 구현에서 배열 또는 행렬 연산에 편리한 메서드들을 제공하고 있어 많이 이용한다.
  • 이중에서 신기한 브로드캐스트 기능에 대해서 설명해보고자 한다.

브로드캐스트

  • 넘파이 연산에서는 서로 형상이 다른 배열끼리의 연산을 가능하게 하는 브로드캐스트라는 기능이 존재한다.
    ex) 배열 [[1,2],[3,4]] 배열과 [10, 20] 배열의 곱연산을 진행해 보겠다.
A = np.array([[1,2],[3,4]])
B = np.array([10, 20])
print(A*B)
// 결과 array([[10,40],[30, 80]])
  • 해당 연산을 시각화 시켜보자

  • 해당 그림처럼 1차원 배열인 B가 똑똑하게 2차원 배열처럼 변환해서 연산을 진행한다.

0개의 댓글