分配的张量切片条件

Question

为什么会这样 '''tensor_ques[1:3,1]=0''' 工作 还有这个 '''tensor_ques[1:3,1] =[0,0]''' 不工作 tensor_ques 是一个 torch.tensor()

我知道 [0,0] 不是张量，那么为什么 =0 可以正常工作

Answer 1

它不起作用，因为[0,0] 是一个列表，而torch 不允许您将列表分配给张量。只有当被分配的项目也是张量时它才会起作用，即tensor_ques[1:3,1] = torch.tensor([0,0])。