对于变量,有时候需要去变量的某一维度的某几个值,可以使用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]])
则有,
对于变量,有时候需要去变量的某一维度的某几个值,可以使用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]])
则有,