pytorch一些基本命令
通过这几天查看R-C3D的详细代码后,我们发现有很多平时没有见过的命令,这里一一记录功能,万一后面会用得上。
import torch
最基本的命令,导入torch库,不然后面的操作都白瞎。
torch.squeeze(a,1)
删除a的第一位维度,对应的有一个unsqueeze()
torch.unsqueeze(a,0)
在a的第0维添加一维
通过这几天查看R-C3D的详细代码后,我们发现有很多平时没有见过的命令,这里一一记录功能,万一后面会用得上。
import torch
最基本的命令,导入torch库,不然后面的操作都白瞎。
torch.squeeze(a,1)
删除a的第一位维度,对应的有一个unsqueeze()
torch.unsqueeze(a,0)
在a的第0维添加一维
本文标题:经过读代码学到的一些pytorch的基本命令操作
本文链接:https://www.haomeiwen.com/subject/ajpagqtx.html
网友评论