摘要: 一、降维 1、求和 对一个n维数组进行求和,如果指定沿哪一个轴来通过求和降低维度。如下是降低0维: A = torch.arange(20, dtype=torch.float32).reshape(5,4) print(A) print(A.sum(axis=0)) /* tensor([[ 0. 阅读全文
posted @ 2024-01-07 18:30 LCAC 阅读(45) 评论(0) 推荐(0)