04 2020 档案

摘要: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 抚琴尘世客 阅读(37428) 评论(0) 推荐(0) 编辑
摘要:1. torch.cat(inputs, dimension=0)说明 torch.cat用于对tensor的拼接,dim默认为0,即从第一维度拼接。表示为4维的图像tensor中,第一维默认为batchSize,第二维为channel(通道),第三维为height(图片的高),第四维为width( 阅读全文
posted @ 2020-04-27 22:08 抚琴尘世客 阅读(2454) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示