PyTorch 求取梯度並實現一元線性迴歸

numpy和tensor相互轉化並求取梯度 *numpy和tensor相互轉化 tensor爲張量,張量指多維矩陣 import torch import numpy as np from torch.autograd import Variable a=torch.Tensor([[2,3],[4,8],[7,9]]) b_=torch.LongTensor([[2,3],[4,8],[7,9
相關文章
相關標籤/搜索