Python 如何使用PyTorch从三维张量中删除元素?

Python 如何使用PyTorch从三维张量中删除元素?,python,pytorch,tensor,Python,Pytorch,Tensor,我有一个张量,形状火炬大小([4161325])。如何删除dim=2上的第一个元素,以使生成的张量具有torch.Size([4161324])?您可以使用简单的切片 >a=torch.randn(4161325) >>>b=a[:,:,1:] >>>b、 形状 火炬尺寸([4161324]) 您可以使用简单的切片 >a=torch.randn(4161325) >>>b=a[:,:,1:] >>>b、 形状 火炬尺寸([4161324]) 进行切片 t=torch.rand(4161325)

我有一个张量,形状
火炬大小([4161325])
。如何删除dim=2上的第一个元素,以使生成的张量具有
torch.Size([4161324])

您可以使用简单的切片

>a=torch.randn(4161325)
>>>b=a[:,:,1:]
>>>b、 形状
火炬尺寸([4161324])

您可以使用简单的切片

>a=torch.randn(4161325)
>>>b=a[:,:,1:]
>>>b、 形状
火炬尺寸([4161324])
进行切片

t=torch.rand(4161325)
t=t[…,1:]#或t=t[省略号,1:]此处,省略号表示DIM的其余部分
t、 形状
火炬尺寸([4161324])
进行切片

t=torch.rand(4161325)
t=t[…,1:]#或t=t[省略号,1:]此处,省略号表示DIM的其余部分
t、 形状
火炬尺寸([4161324])