1、pytorch統計學函數
import torcha = torch.rand(2,2)
print(a)
print(torch.sum(a, dim=0))
print(torch.mean(a, dim=0))
print(torch.prod(a, dim=0))print(torch.argmax(a, dim=0))
print(torch.argmin(a, dim=0))
print(torch.std(a))
print(torch.var(a))
print(torch.median(a))
print(torch.mode(a))a = torch.rand(2,2) * 10
print(torch.histc(a, 6, 0, 0)) #6表示統計多少個區間,0,0是默認a中的最大值最小值a = torch.randint(0, 10, [10])
print(torch.bincount(a)) #只能處理一維的Tensor
2、隨機抽樣
torch.manual_seed(1)
?用于設置隨機數生成器的種子。當你設置了一個種子值,PyTorch 中的所有隨機數生成(在相同的計算設備上)都會變得可確定,即給定相同的種子值,你會得到相同的隨機數序列。
torch.manual_seed(1)
mean = torch.rand(1, 2)
std = torch.rand(1, 2)
print(torch.normal(mean=mean, std=std))