梯度
-
地下城堡4全阵营25位英雄梯度榜与适用场景（梯度.阵营.城堡）
《地下城堡4》正式开启震撼公测！六大阵营齐聚一堂，25位风格迥异的英雄闪亮登场，阵容之丰富令人目不暇接。面对如此多的选择，不少新人玩家陷入选择困难：到底该重点培...
wufei123 发布于 2025-09-24 阅读（12） -
JAX自定义Module梯度计算指南：理解与实践PyTree机制（梯度.自定义.理解.机制.实践...）
本文深入探讨了在JAX中为自定义类（如PyTorch风格的Module）计算梯度时遇到的常见问题及其解决方案。核心挑战在于jax.grad要求被微分函数以参数...
wufei123 发布于 2025-09-24 阅读（11） -
JAX中自定义模块的梯度计算：PyTree注册与函数式转换（梯度.自定义.函数.模块.转换...）
本文深入探讨了在JAX中为自定义Python类（模拟PyTorch Module）计算梯度的核心方法。我们首先识别了直接使用jax.grad对模型输出求导的局...
wufei123 发布于 2025-09-24 阅读（12） -
PyTorch中获取中间张量梯度的实用指南（张量.梯度.获取.实用.指南...）
在PyTorch中，直接通过模块的后向钩子（backward hooks）获取非叶子节点（中间张量）的梯度并非其设计初衷。本文将详细阐述一种有效的方法：通过调...
wufei123 发布于 2025-09-17 阅读（9） -
PyTorch中获取中间张量梯度值的实用指南（张量.梯度.获取.实用.指南...）
本文旨在解决PyTorch反向传播过程中获取非叶子节点（中间张量）梯度的问题。传统的register_backward_hook主要用于模块参数，对中间张量无...
wufei123 发布于 2025-09-17 阅读（9）