随笔分类 - PyTorch / PyTorch库函数学习
摘要：Pytorch保存模型等相关参数，利用`torch.save()`，以及读取保存之后的文件。 ### 函数信息 ```python torch.save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL,_use_new_z
阅读全文
摘要：`torch.multiprocessing.spawn` 是 PyTorch 中用于启动多进程的函数，可以用于分布式训练等场景。其函数签名如下： ```python torch.multiprocessing.spawn( fn, args=(), nprocs=1, join=True, dae
阅读全文
摘要：### 函数简介 Pytorch中的view函数主要用于Tensor维度的重构，即返回一个有相同数据但不同维度的Tensor。 根据上面的描述可知，view函数的操作对象应该是Tensor类型。如果不是Tensor类型，可以通过`tensor = torch.tensor(data)`来转换。 ##
阅读全文
摘要：torch.sum()对输入的tensor数据的某一维度求和，一共两种用法。 方法1详解 torch.sum(input, *, dtype=None) → Tensor input：输入的张量 案例 x = torch.randn(2, 3) print(x) y = torch.sum(x) p
阅读全文
