torch的常用函数

torch是一个开源的机器学习框架，它提供了许多常用的函数来处理张量(Tensors)和进行数学运算。以下是一些常用的torch函数：
1. torch.arange()：返回一个从start开始，以step为步长，到end结束的一维张量。
2. torch.zeros(*sizes, out=None)：返回一个形状为sizes的全0张量。
3. torch.eye(n, m=None, out=None)：返回一个形状为m*n的张量，对角线位置全1，其它位置全0。
4. torch.ones(*sizes, out=None)：返回一个形状为sizes的全1张量。
5. torch.randn(*sizes, out=None)：返回一个形状为sizes的张量，其中的元素是从标准正态分布中随机采样得到的。
6. torch.cat(seq, dim=0, out=None)：将序列seq中的张量沿着指定的维度dim进行拼接。
7. torch.stack(seq, dim=0, out=None)：将序列seq中的张量沿着新的维度dim进行堆叠。
8. torch.sum(input, dim=None, keepdim=False, out=None)：计算输入张量input沿着指定维度dim的和。
9. torch.mean(input, dim=None, keepdim=False, out=None)：计算输入张量input沿着指定维度dim的平均值。
10. torch.max(input, dim=None, keepdim=False, out=None)：计算输入张量input沿着指定维度dim的最大值。
11. torch.min(input, dim=None, keepdim=False, out=None)：计算输入张量input沿着指定维度dim的最小值。