PyTorch实践(二)

Broadcast自动扩展

• 在前面插入一个新的维度
• 扩展维度为1的维度变为相同的维度

tensor拼接与拆分

函数	作用	示例
torch.cat([a ,b],dim=d)	tensor拼接（除d维度其他维度必须相同)
torch.stack([a ,b],dim=d)	创建一个新的维度
torch.split(len, dim=d)	按长度将tensor拆分
torch.chunk(num,dim=d)	按数量将tensor进行拆分

数学运算

函数	说明	示例
torch.matmul(a,b)	a,b相乘(最后两位)
a@b	a,b相乘	只适用于2d
torch.clamp(min,max)	小于min为min，大于max为max

属性统计

函数	说明	示例
torch.norm(d)	得到矩阵的范数
torch.topk(n,dim=d)	最大的k个值
torch.kthvalue(n,dim=d)	返回第k小的值和索引

|

高阶操作

函数	说明	示例
torch.where(condition,x,y)