2021年8月11日

torch.Tensor 的一些注意点

摘要: 1. 广播机制 2. expand和repeat 3. view和reshape 1. 广播机制 广播机制牢记一点就可以了:从末尾开始连续找第一个不相同的维度容量开始扩充,直到扩充成维度相同。 比如: a = torch.rand(3, 4)b = torch.rand(4)c1 = a + b # 阅读全文

posted @ 2021-08-11 10:25 博闻强记2010 阅读(221) 评论(0) 推荐(0) 编辑

导航