PyTorch中Tensor的學習筆記

1 緻謝

感謝PyTorch官方提供的文檔，

連結：https://pytorch.org/docs/stable/tensors.html

2 如何建立一個Tensor

2.1 使用torch.tensor()直接建立Tensor

示例代碼如下：

torch.tensor([[1, 2, 3], [4, 5, 6]])
           

2.2 如何建立作為參數的Tensor

（使用預設方法建立的Tensor，屬性requires_grad=False）

示例代碼如下：

x = torch.tensor([[1.], [-1.]], requires_grad=True)
           

3 PyTorch中Tensor的索引

3.2 可以使用boolTensor對Tensor進行“過濾”

boolTensor對Tensor有“過濾”的效果，示例的效果下圖所示，