ydy2511000的博客

本文是记录自己学习中的一些小知识，比较散 a = torch.randn(4, 3, 28, 28) print(a[:2].shape) #out: torch.Size([2, 3, 28, 28]) a[:2] 这里是指，a的第0维中，0到2，不包含2的数据，其他维度默认取全部，等效于a[:2, :, :, :] 让然，很明显，a[:, :, :, :]， 这个能控制自己想要去...

.floor() 往下取整数 .ceil() 网上取整数 .round() 四舍五入 >=0.5向上取整，<0.5向下取整 .trunc() 裁剪，只取整数部分 .frac() 只取小数部分 如何查看梯度的模 w.grad.norm(2) 梯度的二范数，可以用于观测梯度是否爆炸（一般都比较小，10左右，100都算比较大了） 梯度裁剪时，会用到，gradient cl...