weixin_42028608的博客

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...

Tutorials1. Generaldata loading：讲自定义数据加载操作iteratorpipeline构建DALI expressions and arithmetic operations：: 讲可以在tensor上自定义一些+ - * /操作Multiple GPU support：主要讲多GPU（shard_id：显卡id, num_shards：将数...