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