nonzero()

以张量的形式返回非0元素的索引，不管有几维，维度索引都放在一个“列表”中

import torch

a =torch.Tensor([1,2,4,4,5])
print(torch.nonzero(a<4))

torch.unique()

挑出tensor中的独立不重复元素。

# pytorch # python
Logo
https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

cover

GitHub Copilot 学生认证详细教程

avatar AI编程社区
cover

AI编程：做个小工具

avatar AI编程社区

Java Web 开发环境（JDK+Tomcat+Maven）安装配置全攻略

Maven 运行依赖 JDK，需先确保你已安装 JDK 1.8 及以上版本，且JAVA_HOME环境变量配置完成（可通过cmd输入java -version验证）。：无中文、无空格、无特殊字符（示例：D:\jdk1.8.0_102，禁止D:\编程工具\JDK/D:\jdk 1.8）。：无中文、无空格、无特殊字符（示例：D:\jdk1.8.0_102，禁止D:\编程工具\JDK/D:\jdk 1.8

avatar AI编程社区