您好,欢迎光临本网站![请登录][注册会员]  

搜索资源列表

  1. 基于PyTorch的permute和reshape/view的区别介绍

  2. 主要介绍了基于PyTorch的permute和reshape/view的区别介绍,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
  3. 所属分类:其它

    • 发布日期:2020-09-16
    • 文件大小:47kb
    • 提供者:weixin_38707153
  1. 基于PyTorch的permute和reshape/view的区别介绍

  2. 二维的情况 先用二维tensor作为例子,方便理解。 permute作用为调换Tensor的维度,参数为调换的维度。例如对于一个二维Tensor来说,调用tensor.permute(1,0)意为将1轴(列轴)与0轴(行轴)调换,相当于进行转置。 In [20]: a Out[20]: tensor([[0, 1, 2], [3, 4, 5]]) In [21]: a.permute(1,0) Out[21]: tensor([[0, 3], [1, 4], [2, 5]])
  3. 所属分类:其它

    • 发布日期:2020-12-17
    • 文件大小:50kb
    • 提供者:weixin_38703823