Pytorch学习日志之函数用法记录

1.torch.max() return 所有元素中的最大值.

   torch.max(input,dim,keepdim=False),dim =0时返回每一列的最大值及索引,dim=1时返回每一行的最大值及索引。

    keepdim=True时返回的值的形状与dim相同

2.torch.squeeze(a)压缩第a维维度为1的轴,torch.squeeze(),压缩所有维度为1的轴,torch.unsqueeze(a),在第a个轴上扩充一个维度为1的轴

3.torch.Tensor.view(),根据原有数据重新分配为一个新的张量,eg: x = torch.randn(2,4),y=x.view(4,2)

  torch.Tensor.view_as(tensor)相当于torch.Tensor.view_as(tensor.size())


版权声明:本文为qq_40103460原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。