如何计算N-d张量中的非零行？

import numpy as np

def find_nonzero_rows(tensor):
    non_zero_rows = []
    for i in range(tensor.shape[0]):
        row = tensor[i]
        if np.count_nonzero(row) > 0:
            non_zero_rows.append(i)
    return non_zero_rows

# 示例使用
tensor = np.array([[1, 0, 2], [0, 0, 0], [3, 4, 0]])
non_zero_rows = find_nonzero_rows(tensor)
print(non_zero_rows)