shaoheshaohe 发表于 2019-4-21 21:13:21

Pytorch中tensor常用语法

上次我总结了在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清子 发表于 2019-5-11 14:34:40

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

Happy清子 发表于 2019-5-11 14:35:32

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

Happy清子 发表于 2019-5-11 14:35:56

理解tensor是
AI编程的基础
页: [1]
查看完整版本: Pytorch中tensor常用语法