04 2020 档案
摘要:1. torch.sum(input, dim, out=None) 参数说明: input:输入的tensor矩阵。 dim:求和的方向。若input为2维tensor矩阵,dim=0,对列求和;dim=1,对行求和。注意:输入的形状可为其他维度(3维或4维),可根据dim设定对相应的维度求和。
阅读全文
摘要:1. torch.cat(inputs, dimension=0)说明 torch.cat用于对tensor的拼接,dim默认为0,即从第一维度拼接。表示为4维的图像tensor中,第一维默认为batchSize,第二维为channel(通道),第三维为height(图片的高),第四维为width(
阅读全文