Day2:torch的数组运算

import torch

a=torch.rand(5,6)
print(a)
print(torch.mean(a))##求所有元素的平均值
print(torch.sum(a))##求所有元素的总和
print(torch.prod(a))##求所有元素相乘
print(torch.mean(a, dim=0))##求每一列的平均值
print(torch.mean(a, dim=1))##求每一行的平均值
print( torch.sum(a, dim=0))##求每一列的总数
print( torch.sum(a, dim=1))##求每一行的总数
print(torch.prod(a, dim=0))##求每一列所有元素的乘积
print(torch.prod(a, dim=1))##求每一行所有元素的乘积
print(torch.argmax(a, dim=0))##求每一列的最大值的索引值
print(torch.argmax(a, dim=1))##求每一行的最大值的索引值
print( torch.argmin(a, dim=0))##求每一列的最小值的索引值
print( torch.argmin(a, dim=1))##求每一行的最小值的索引值

b=torch.randint(1,10,[10])
print(b)
print(torch.bincount(b))##返回每个值的频数
# 本例输出的结果就是统计0 出现多少次，1 出现多少次，2 出现多少次。。。。。。
# 分别是对应位置的统计次数，比如输出的列表中的第一个位置就是0出现了多少次
# 另外bincount只能处理一维的数据