2020 年 6月 3 日随笔档案 - jaysonteng

2020年6月3日

摘要： 1、加减乘除 a + b = torch.add(a, b) a - b = torch.sub(a, b) a * b = torch.mul(a, b) a / b = torch.div(a, b) import torch a = torch.rand(3, 4) b = torch.ran 阅读全文

posted @ 2020-06-03 22:49 jaysonteng 阅读(12209) 评论(0) 推荐(1)

pytorch 数据拼接与拆分cat、stack、split、chunck

摘要： 1、cat拼接功能：通过dim指定维度，在当前指定维度上直接拼接默认是dim=0 指定的dim上，维度可以不相同，其他dim上维度必须相同，不然会报错。 1）拼接两个维度相同的数 a = torch.rand(2, 3, 2) a # 输出： tensor([[[0.6072, 0.6531], 阅读全文

posted @ 2020-06-03 15:58 jaysonteng 阅读(1700) 评论(0) 推荐(0)