(a, b, ... n, n + 1, ... m) 크기의 텐서와 (n, n + 1, ... m) 크기의 텐서 사이에 브로드캐스팅으로 원소별 연산을 적용할 수 있습니다. 이때 브로드캐스팅은 a부터 n - 1까지의 축에 자동으로 일어납니다.
다음은 크기가 다른 두 텐서에 브로드캐스팅으로 원소별 maximum 연산을 적용하는 예입니다.
import numpy as np x = np.random.random((64, 3, 32, 10)) ➊ y = np.random.random((32, 10)) ➋ z = np.maximum(x, y) ➌
➊ x는 (64, 3, 32, 10) 크기의 랜덤 텐서입니다.
➋ y는 (32, 10) 크기의 랜덤 텐서입니다.
➌ 출력 z 크기는 x와 동일하게 (64, 3, 32, 10)입니다.