torch 常用函数

 

norm 范式
import numpy as np
np_data=[1,2]
torch_data = torch.from_numpy(np.asarray(np_data).astype(np.float32))
# x=torch.Tensor({{1,2,3,4}, {5,6,7,8}})
y = torch.norm(torch_data, 2)  # p-范式
print(y)

div 除法

import numpy as np
np_data=[1,2]
torch_data = torch.from_numpy(np.asarray(np_data).astype(np.float32))
# x=torch.Tensor({{1,2,3,4}, {5,6,7,8}})
y = torch.div(torch_data,2)  # 除以2
print(y)

按照后者维度扩展
expand_as