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)

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/81476407
今日推荐