pytorch
关注
分享
扫一扫
码代码-是个回文串
这个作者很懒，什么都没留下…
展开
-
torch.distributions.Categorical
主要就是根据概率来产生指定shape的分布原创 2021-04-20 19:38:57 · 1594 阅读 · 0 评论 -
torch.broadcast_tensors(*tensors)
没看懂原创 2019-11-15 10:14:23 · 1257 阅读 · 4 评论 -
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → bool
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → bool 在一定的误差允许内，判断两个值是否相等原创 2019-10-30 16:29:07 · 2620 阅读 · 0 评论 -
torch.mean
torch.mean(input, dim, keepdim=False, out=None) → Tensor **其中，keepdim的参数很重要，如果使用默认的False，则默认降维，如果使用True，就不进行降维 **原创 2019-10-30 15:05:54 · 746 阅读 · 0 评论