一个形状为[2500,256]的张量，判断2500个向量中哪个为全0

import numpy as np

# 假设 your_tensor 是你的 [2500, 256] 形状的张量
your_tensor = np.random.randn(2500, 256)  # 这里用随机数生成张量作为示例

# 判断每个子张量是否全为零
is_zero = np.all(your_tensor == 0, axis=1)

# 找出所有全零张量的索引
zero_indices = np.where(is_zero)[0]

print(zero_indices)
 

OR

import torch

# 假设 your_tensor 是你的 [2500, 256] 形状的张量
your_tensor = torch.randn(2500, 256)  # 这里用随机数生成张量作为示例

# 判断每个子张量是否全为零
is_zero = torch.all(your_tensor == 0, dim=1)

# 找出所有全零张量的索引
zero_indices = torch.where(is_zero)[0]

print(zero_indices)