torch.narrow()
用法:torch.narrow(input, dim, start, length) → Tensor
返回输入张量的切片操作结果。 输入tensor和返回的tensor共享内存。
input (Tensor) – 需切片的张量
dim (int) – 切片维度
start (int) – 开始的索引
length (int) – 切片长度
torch.unbind()
移除指定维后,返回一个元组,包含了沿着指定维切片后的各个切片。
用法:torch.unbind(input, dim=0) → seq
返回指定维度切片后的元组。
dim=-1表示按照最后一个维度切片,切片之后维度会少1
torch.stack
沿着一个新维度对输入张量序列进行连接。 序列中所有的张量都应该为相同形状。
用法:outputs = torch.stack(inputs, dim=?) → Tensor
inputs : 待连接的张量序列。(list和tuple)
dim : 新的维度