05 2022 档案

摘要:广播机制 从后往前依次检查维度,如果两个张量对应的维度上数目相等,则会按照该维度相加 若其中一个维度数目为1,则会应用广播机制. 如: a = torch.arange(3).reshape([1, 3, 1]) b = torch.arange(3).reshape([1, 3, 1]) a + 阅读全文
posted @ 2022-05-03 20:45 Krasitz 阅读(172) 评论(0) 推荐(0)