当前位置：网站首页>[pytorch] common functions

[pytorch] common functions

2022-04-21 22:22:00 【Coke Daniel】

Here's the catalog title

tensor

tensor

establish ：
Insert picture description here
Get shape ：shape perhaps size function
Add ：torch.add(x,y,out=y), There are other ways, of course , But this is the best way ！
Indexes ： And numpy similar
Other selection functions ：

Change shape ：view function ,1、 And you can use -1 Indicates that the system automatically calculates ,x=torch.randn(15).view(-1,5);2、view Back to tensor Original tensor Shared memory , You can start with clone In the use of view,y=x.clonew().view(15).
take tensor Convert to python Of number：item function
Line generation function ：
Insert picture description here
tensor turn numpy：tensor.numpy()
numpy turn tensor：torch.from_numpy()
to function take Tensor You put it on the device