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
posted @ 2021-08-11 10:25