索引,切片,连接和转换操作续
torch.narrow(input, dim, start, length) → Tensor
根据指定的维度,维度的start和长度,返回一个新的张量 参数
input...tensor([ 0, 1, 2, 3])
torch.split(tensor, splitsizeor_sections, dim=0)
把张量分割为块, 如果splitsizeorsections是整数类型...>>> y.size()
torch.Size([2, 2, 1, 2])
torch.stack(tensors, dim=0, out=None) → Tensor
沿着一个新维度连接张量...,张量的形状需要一样 参数
tensors(张量列表)
dim(int):插入的维度,必须在0和连接张量的维度之间
out