pytorch 的 sum 和 softmax 方法 dim 参数的使用
摘要：
在阅读使用 pytorch 实现的代码时，笔者会遇到需要对某一维数据进行求和( sum )或 softmax 的操作。在 pytorch 中，上述两个方法均带有一个指定维度的 dim 参数，这里记录下 dim 参数的用法。 torch.sum 在 pytorch 中，提供 torch.sum 的两种 阅读全文
posted @ 2020-04-15 20:20 yhjoker 阅读(8323) 评论(0) 推荐(1) 编辑
