2020年4月29日
Pytorch tensor求和（ tensor.sum()）
摘要： 1. torch.sum(input, dim, out=None) 参数说明： input：输入的tensor矩阵。 dim：求和的方向。若input为2维tensor矩阵，dim=0，对列求和；dim=1，对行求和。注意：输入的形状可为其他维度（3维或4维），可根据dim设定对相应的维度求和。 阅读全文
posted @ 2020-04-29 09:52 抚琴尘世客 阅读(37743) 评论(0) 推荐(0)