基本配置导入包和版本查询import torch import torch.nn as nn import… ... torch.nonzero(tensor) # index of non-zero elements torch.nonzero(tensor==0) # index ...
確定! 回上一頁