摘要: 结论: torch.max(),不指定dim参数,返回输入张量中所有数据的最大值; 如果指定dim参数,则返回按照指定维度的最大值和各最大值对应的位置 torch.argmax(),不指定dim,然后输入张量中所有数据最大值位置(按照张量被拉伸为1为向量算); 如果指定dim,返回指定维度的最大值位 阅读全文
posted @ 2021-07-28 17:10 哈哈哈喽喽喽 阅读(398) 评论(0) 推荐(0) 编辑
摘要: ** 结论 torch.repeat: 输入张量的从后往前的后面维度对应按照repeat中大小进行repeat操作(所以 输入张量维度>= repeat维度)。 假设输入张量为(a,b,c),repeat(x,y),则为b维度repeat x倍,c维度repeat y倍;最终输出维度为(a, bx, 阅读全文
posted @ 2021-07-28 11:12 哈哈哈喽喽喽 阅读(2313) 评论(0) 推荐(0) 编辑