1. 程式人生 > 其它 >numpy與torch相同功能命令

numpy與torch相同功能命令

技術標籤:pytorch

1.資料生成

0向量

import torch
import numpy as np

a1=torch.zeros(10,1)
a2=np.zeros((10,1))

2.求範數

torch.norm(x,p,dim)

np.linalg.norm(x,p) 預設2範數 np.linalg.norm(x) 等於 sqrt(np.sum(x**2))