Pytorch 中的 memory_format 参数 与 channels_last

  • memory_format(torch.memory_format, optional):期望返回的tensor的内存格式,默认为torch.preserve_format.

memory_format = torch.channels_last
  • PyTorch 1.5 引入了channels_last对卷积网络内存格式的支持。此格式旨在与 AMP结合使用,以进一步加速具有 Tensor Cores的卷积神经网络

你可能感兴趣的:(Pytorch,中的各种函数,Pytorch)