Pytorch中tensor常用语法

shaoheshaohe

上次我总结了在PyTorch中建立随机数Tensor的多种方法的区别。

这次我把常用的Tensor的数学运算总结到这里，以防自己在使用PyTorch做实验时，忘记这些方法应该传什么参数。

总结的方法包括：

Tensor求和以及按索引求和：torch.sum() torch.Tensor.indexadd()

Tensor元素乘积：torch.prod(input)

对Tensor求均值、方差、极值：

torch.mean() torch.var()

torch.max() torch.min()

最后还有在NLP领域经常用到的：

求Tensor的平方根倒数、线性插值、双曲正切

torch.rsqrt(input) torch.lerp(star,end,weight)

torch.tanh(input, out=None)

Happy清子

tensor， img，array三者之间的互换

Happy清子

python的灵活性
在这里
淋漓尽致地体现了

Happy清子

理解tensor是
AI编程的基础