2021年10月8日
6 Tensor.squeeze & Tensor.unsqueeze
摘要： Tensor.squeeze & Tensor.unsqueeze 1 Tensor的维度 张量的的定义：一个 n 维的张量就是一维数组中的所有元素都是 n - 1 维的张量。 举例说明： import torch a = torch.rand(3) b = torch.rand(2, 3) c = 阅读全文
posted @ 2021-10-08 15:43 SethDeng 阅读(241) 评论(0) 推荐(0) 编辑