Pytorch 基本統計量

平均値などの基本統計量を、Pytorchで。

1. サンプル

import torch

#-- floatで指定
x = torch.tensor([[1.0,2.0,3.0]])

#-- 平均値
print(torch.mean(x).item())

#-- 合計値
print(torch.sum(x).item())

#-- 最大値
print(torch.max(x).item())

#-- 最小値
print(torch.min(x).item())

2.実行結果

2.0
6.0
3.0
1.0