pytorch 基本运算函数

生成随机数：
torch.rand() torch.randn()
tensor乘法：
torch.mm(tensor1, tensor2)
返回tensor最大值， 前K个最大值：
value, index = torch.max(input, dim) value, index = torch.topk(input, k, dim)
tensor拼接：
torch.cat(list, index)
tensor转置：
torch.Tensor.permute(0,1,2,3,4)
unsqueeze：添加维度 [3,4] -> [1,3,4]
torch.Tensor.unsqueeze(dim) torch.unsqueeze(tensor, dim)
squeeze：去掉维数为1的维度 [1,3,4] -> [3,4]
torch.Tensor.squeeze() torch.squeeze(tensor)
 

 

 

 

 

 

 

• « 上一篇：JVM性能调优监控工具jps、jstack、jmap、jhat、jstat、hprof使用详解
• » 下一篇：vue对storejs获取的数据进行处理时遇到的问题