摘要: `torch.multiprocessing.spawn` 是 PyTorch 中用于启动多进程的函数,可以用于分布式训练等场景。其函数签名如下: ```python torch.multiprocessing.spawn( fn, args=(), nprocs=1, join=True, dae 阅读全文
posted @ 2023-06-26 19:08 张Zong在修行 阅读(3521) 评论(0) 推荐(0) 编辑
摘要: 正则表达式中,`group()`用来提出分组截获的字符串,`()`用来分组。 **例如:** ```python import re a = "123abc456" print re.search("([0-9]*)([a-z]*)([0-9]*)",a).group(0) #123abc456,返 阅读全文
posted @ 2023-06-26 17:07 张Zong在修行 阅读(1294) 评论(0) 推荐(0) 编辑
摘要: 首先导入依赖的torch包。 ```python import torch ``` 我们设: + seq_len(序列的最大长度):5 + batch_size(批量大小):2 + d_model(每个单词被映射为的向量的维度):10 + heads(多头注意力机制的头数):5 + d_k(每个头的 阅读全文
posted @ 2023-06-26 13:29 张Zong在修行 阅读(768) 评论(0) 推荐(0) 编辑
摘要: ### 函数简介 Pytorch中的view函数主要用于Tensor维度的重构,即返回一个有相同数据但不同维度的Tensor。 根据上面的描述可知,view函数的操作对象应该是Tensor类型。如果不是Tensor类型,可以通过`tensor = torch.tensor(data)`来转换。 ## 阅读全文
posted @ 2023-06-26 10:07 张Zong在修行 阅读(478) 评论(0) 推荐(0) 编辑