Numpy
- 随机矩阵:
np.random.randn(d0, d1, d2, ...)
Pytorch
- 随机矩阵:
torch.randn(d0, d1, d2, ...)
- 添加维度:
tensor.unsqueeze(0)
- 压缩维度:
tensor.squeeze(0)
- 按维度拼接tensor:
torch.cat(inputs, dim=0, ...)
- 维度堆叠:
torch.stack(inputs, dim=0)
- 张量排序索引:
tensor.sort(descending=True)
返回一个tensor为排序后的tensor, 一个为index_tensor - 矩阵元素夹逼:
tensor.clamp()
- s