qq_29494693的博客

代码 import torch x = torch.randn(3,4) print(x) print(torch.mean(x,dim=1,keepdim=True)) mz_items = x[:,] > torch.mean(x,dim=1,keepdim=True) print(mz_items) import numpy as np mz_items_np = mz_items.numpy() index = np.argwhere( mz_items_np == True ) print(