![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 78
一只帅气的小菜鸡
这个作者很懒,什么都没留下…
展开
-
Pytorch常用总结(持续更新...)
Pytorchdim用法:参考:https://blog.csdn.net/qq_27261889/article/details/88613932### testa = torch.rand((3,4))print(a.size())print(a)b = torch.argmax(a, dim=1) #意味着以行为单位操作所有列,从每一行的所有列中找最大print(b)pr...原创 2020-05-03 21:13:10 · 308 阅读 · 0 评论 -
【Pytorch】nvidia-dali——一种加速数据增强的方法
Tutorials1. Generaldata loading:讲自定义数据加载操作iteratorpipeline构建DALI expressions and arithmetic operations:: 讲可以在tensor上自定义一些+ - * /操作Multiple GPU support:主要讲多GPU(shard_id:显卡id, num_shards:将数...原创 2020-05-03 10:19:40 · 7975 阅读 · 7 评论