2021-04-02
Torch TensorTensor类型创建Tensor索引与切片维度变换合并与分割
Tensor类型
a = torch.randn(2,3)
a.type() #查看类型
isinstance(a, torch.FloatTensor) # 判断类型
创建Tensor
属性:tensor.numel() 元素数量, tensor.dim() 维度, tensor.shape() 具体的形状,
tensor.size(0) == tensor.shape[0]
a = torch.rand(
原创
2021-04-02 16:50:57 ·
85 阅读 ·
0 评论