The simplest and best solution is to use torch.sum() . ... Alternatively, you can use tensor.sum(axis) where axis indicates 0 and 1 for summing over rows and ...
確定! 回上一頁