美文网首页
pytorch 维度计算

pytorch 维度计算

作者: 潘旭 | 来源:发表于2020-05-21 10:43 被阅读0次

    维度计算

    import torch
    
    x = torch.tensor([[1, 2, 3], [4, 5, 6]])
    
    print(f"x shape: {x.shape}, shape 0: {x.shape[0]}, shape 1: {x.shape[1]}")
    print(f"x size: {x.size()}, size 0: {x.size(0)}, size 1: {x.size(1)}")
    print(f"x dim: {x.dim()}, len(size): {len(x.size())}, len(shape): {len(x.shape)}")
    
    x shape: torch.Size([2, 3]), shape 0: 2, shape 1: 3
    x size: torch.Size([2, 3]), size 0: 2, size 1: 3
    x dim: 2, len(size): 2, len(shape): 2
    

    关于 shape 和 size() 的区别, 二者是一样的,区别在于 shape 是为了和numpy对应的,另外 在获取不同的维度的时候不太一样. dim 是获取实际的维度,相当于 len(x.shape) 或者 len(x.size())

    相关文章

      网友评论

          本文标题:pytorch 维度计算

          本文链接:https://www.haomeiwen.com/subject/yufzohtx.html