PyTorch一些有趣而又实用的小操作
PyTorch有趣而又实用的小操作如何取出高维张量中满足一定条件的值(比如大于0.5),其余设为零?解决方案一,张量花式索引解决方案二,torch.where()的API如何取出高维张量中满足一定条件的值(比如大于0.5),其余设为零?解决方案一,张量花式索引 代码如下:a = t.randn([2,3])print(a)a[a<0.5] = 0print(a) 结果:...
·
更多推荐
已为社区贡献1条内容
所有评论(0)