-
- Evaluate Expression
调试过程中，需要查看比如attn_weight 张量tensor的值。
- Evaluate Expression
-
- 方法一：
attn_weight.detach().numpy(),view as array
- 方法一：
-
- 方法二：
attn_weight.cpu().numpy(),view as array
- 方法二：
pycharm pytorch tensor张量可视化，view as array
取个名字真难呐2025-01-04 21:46
相关推荐
阿斌.com2 小时前
Visual Studio 2026安装全攻略头发够用的程序员3 小时前
TensorRT 自定义算子插件实战（一）：从零手写 customScaledTanh论文复现现场4 小时前
RTX 5090 32GB 适合科研计算吗？单卡训练、推理、论文复现与 OOM 判断方法论文复现现场6 小时前
论文复现看到“RTX 3090 or higher”：显存、CUDA、batch size 与 OOM 怎么判断？我要神龙摆尾9 小时前
IDEA 阅读插件zh路西法13 小时前
【ResNet18】从图像到视觉特征：卷积神经网络如何理解图像梦幻通灵13 小时前
IDEA 实用快捷键【持续更新】FPGA小徐13 小时前
Ubuntu 22.04 安装 Miniconda、PyTorch 与 YOLOv8 并完成 CPU 推理CAE虚拟与现实1 天前
PyTorch和scikit-learn的区别