pytorch
码代码-是个回文串
这个作者很懒,什么都没留下…
展开
-
torch.distributions.Categorical
主要就是根据概率来产生指定shape的分布原创 2021-04-20 19:38:57 · 1575 阅读 · 0 评论 -
torch.broadcast_tensors(*tensors)
没看懂原创 2019-11-15 10:14:23 · 1241 阅读 · 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 · 2585 阅读 · 0 评论 -
torch.mean
torch.mean(input, dim, keepdim=False, out=None) → Tensor**其中,keepdim的参数很重要,如果使用默认的False,则默认降维,如果使用True,就不进行降维**原创 2019-10-30 15:05:54 · 730 阅读 · 0 评论