判断是否有 nan

torch.any(torch.isnan(a))

利用 torch.where() 函数替换所有 nan

where() 函数有三个输入值，第一个是判断条件，第二个是符合条件的设置值，第三个是不符合条件的设置值。

a = torch.Tensor([[1, 2, np.nan], [2, np.nan, 4], [3, 4, 5]])

a = torch.where(torch.isnan(a), torch.full_like(a, 0), a)
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

Claude Code 使用技巧

Claude Code 使用摘要 Claude Code 提供三种交互模式（默认/自动接受/计划模式），支持多种快捷键和斜杠命令管理对话、记忆和任务。用户可通过CLI启动，使用!执行Shell命令，利用Skill复用常用指令，并通过Subagents处理独立任务。记忆系统分为项目级和用户级，支持图片输入和Hooks自动化。MCP协议可扩展外部工具集成，插件系统增强功能边界。

avatar 2048 AI社区

Babel幽灵注释：删节点为何删不掉注释？

Babel中"幽灵注释"问题的核心原因是注释并非节点的属性，而是通过leadingComments/trailingComments关联的独立对象。当使用path.remove()删除节点时，注释对象及其位置信息仍保留在内存中，导致生成代码时注释被错误保留或漂移。解决方案包括：1)删除前手动清空注释引用；2)用空语句替换节点；3)清除位置元数据。理解Babel"宁留勿漏"的设计哲学，按照"清注释→

avatar 2048 AI社区

AI Agent Harness Engineering 算力优化：边缘部署与云端协同的成本控制技巧

边缘计算是一种分布式计算范式，它将计算任务、数据存储和应用服务放在靠近用户/设备的“网络边缘”——这里的“网络边缘”是指距离数据产生或消费的位置不超过“一跳”或“几跳”的节点（例如，用户的手机/PC、家庭路由器、IoT网关、5G基站的UPF（用户面功能）节点、零售门店的本地服务器、工厂车间的边缘控制器等）。核心属性维度（我们将在2.1.3用Markdown表格详细对比）：延迟（Latency）：端

avatar 2048 AI社区