本文主要是介绍torch narrow用法,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
对于变量,有时候需要去变量的某一维度的某几个值,可以使用narrow函数实现,例如对于变量x,narrow函数输入为,
`x.narrow`(*dimension*, *start*, *length*) → Tensor
表示取变量x的第dimension维,的从索引start开始到,start+length范围的值.
例如,我们取变量x的值为,
x = torch.Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
则有,
这篇关于torch narrow用法的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!