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有“过滤”的效果,示例的效果下图所示,

猜你喜欢

转载自blog.csdn.net/songyuc/article/details/106662741