pytorch中实现与np.dot类似功能

定义tensor数组时使用dtype=torch.float , 统一成float格式
即可矩阵内积(类似 np.dot(x, W) )
y = torch.matmul(x, W)

你可能感兴趣的:(python)