Normalize to sum 1. /sum(A, dims = 2) sum(sum(A, dims =2) .