torch 基础

[文章目录]

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 : 新的维度

发表评论

邮箱地址不会被公开。 必填项已用*标注