torch.count_nonzero. torch.count_nonzero(input, dim=None) → Tensor. Counts ... optional) – Dim or tuple of dims along which to count non-zeros. Example:
確定! 回上一頁