2020年7月4日
assert
摘要： assert len(x.shape) in (2, 4) 阅读全文
posted @ 2020-07-04 20:32 6+0 阅读(191) 评论(0) 推荐(0) 编辑
numpy中的axis和Pytorch中的dim参数
摘要： 1、二维： axis=0表示分别对每一列做运算； axis=1表示分别对每一行做运算。 2、高维： 假设输入的形状是(m, n, k): -如果指定axis(dim)=0, 输出的size就是(1, n, k)或者(n, k) -如果指定axis(dim)=1, 输出的size就是(m, 1, k) 阅读全文
posted @ 2020-07-04 20:26 6+0 阅读(756) 评论(0) 推荐(0) 编辑