python 统计矩阵元素

Posted 2023-02-27 AI视觉网奇

目录

pytorch tensor统计元素个数：

numpy统计矩阵

---

pytorch tensor统计元素个数：

import torch
import numpy as np

def test():
    torch.manual_seed(1)
    a = torch.randn(3, 4)
    print(a)
    print(a.size())
    print(*a.size())
    print(np.multiply(*a.size()))

test()

结果：
tensor([[ 0.6614,  0.2669,  0.0617,  0.6213],
        [-0.4519, -0.1661, -1.5228,  0.3817],
        [-1.0276, -0.5631, -0.8923, -0.0583]])
torch.Size([3, 4])
3 4
12

原文链接：https://blog.csdn.net/qq_43678005/article/details/122666046

numpy统计矩阵

    import numpy as np

    a = np.array([[3, 7, 5], [8, 4, 3], [2, 4, 9]])

    print(a.size)
    print(a.shape)

结果：

9
(3, 3)

numpy统计元素出现个数：

    test1 = np.array([[0, 1, 0], [0, 1, 1], [0, 1, 2]])
    x = np.bincount(test1.flatten())
    print(x)