索引,切片,连接和转换操作续
torch.narrow(input, dim, start, length) → Tensor
根据指定的维度,维度的start和长度,返回一个新的张量 参数
input...,如果可以,会返回一个input的view。...):
dim:沿着分割的维度
torch.squeeze(input, dim=None, out=None) → Tensor
返回一个张量,移除所有size为1的维度 举个例子,比如输入的形状是 (...>>> y.size()
torch.Size([2, 2, 1, 2])
torch.stack(tensors, dim=0, out=None) → Tensor
沿着一个新维度连接张量...,张量的形状需要一样 参数
tensors(张量列表)
dim(int):插入的维度,必须在0和连接张量的维度之间
out