Concept
Broadcasting with Non-Reduction Summation
Because a non-reduction sum retains its original axes after summing along a specific dimension, the resulting tensor can be directly used in elementwise operations with the original tensor via broadcasting. For instance, dividing a matrix by its non-reduction row sum seamlessly triggers the broadcasting mechanism to expand the sum tensor, thereby generating a new matrix where the elements of each row mathematically sum up to .
0
1
Updated 2026-05-03
Tags
D2L
Dive into Deep Learning @ D2L