zoukankan      html  css  js  c++  java
  • Pytorch 一些函数用法

    PyTorch中view的用法:https://blog.csdn.net/york1996/article/details/81949843

    max用法

    import torch
    
    d=torch.Tensor([[1,3],[2,4]])
    print(d)

    tensor([[1., 3.],
            [2., 4.]])

    import torch
    
    d=torch.Tensor([[1,3],[2,4]])
    print(torch.max(d,0))
    print(torch.max(d,1))

    (tensor([2., 4.]), tensor([1, 1]))
    (tensor([3., 4.]), tensor([1, 1]))

    torch.max()返回两个结果,第一个是最大值,第二个是对应的索引值;第二个参数 0 代表按列取最大值并返回对应的行索引值,1 代表按行取最大值并返回对应的列索引值。

  • 相关阅读:
    if判断语句和循环语句
    列表,元祖,字典的详细概述
    day10
    day09
    day08
    java---基本程序设计总结
    day07
    day06
    day05
    day04
  • 原文地址:https://www.cnblogs.com/cunyusup/p/9552555.html
Copyright © 2011-2022 走看看