pytorch入門- tensor基礎操做(1)

建立tensor 建立指定元素的tensor: torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) 建立隨機元素的tensor: torch.rand(*size, out=None, dtype=None, layout=torch.strided, device=None, req
相關文章
相關標籤/搜索