请选择 进入手机版 | 继续访问电脑版
查看: 179|回复: 3

Pytorch中tensor常用语法

[复制链接]

326

主题

666

帖子

4179

积分

xdtech

Rank: 5Rank: 5

积分
4179
发表于 2019-4-21 21:13:21 | 显示全部楼层 |阅读模式
上次我总结了在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)

回复

使用道具 举报

130

主题

511

帖子

5319

积分

xdtech

Rank: 5Rank: 5

积分
5319
发表于 2019-5-11 14:34:40 | 显示全部楼层
tensor, img,array三者之间的互换
回复

使用道具 举报

130

主题

511

帖子

5319

积分

xdtech

Rank: 5Rank: 5

积分
5319
发表于 2019-5-11 14:35:32 | 显示全部楼层
python的灵活性
在这里
淋漓尽致地体现了
回复

使用道具 举报

130

主题

511

帖子

5319

积分

xdtech

Rank: 5Rank: 5

积分
5319
发表于 2019-5-11 14:35:56 | 显示全部楼层
理解tensor是
AI编程的基础
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表