import torch

sample = torch.rand(3, 2)

# 求和
print(torch.sum(sample))

# 求最小值
print(torch.min(sample))

# 求最大值
print(torch.max(sample))

# 求最小值所在位置
print(torch.argmin(sample))

# 求最大值的位置
print(torch.argmax(sample))

# 求平均值
print(torch.mean(sample))

# 求中位数
print(torch.median(sample))

# 计算开方
print(torch.sqrt(sample))

# 求平方
print(sample ** 2)