pytorch中张量形状怎么看
时间: 2023-12-19 07:29:51 浏览: 92
Pytorch中文文档.docx
在 PyTorch 中,可以通过张量的 `shape` 属性查看其形状。例如,对于一个名为 `tensor` 的张量,可以通过 `tensor.shape` 来获取其形状。这个属性返回一个元组,元组的每个元素表示该维度上的元素个数。例如,对于一个形状为 (3, 4, 5) 的张量,其 `shape` 属性返回的元组为 `(3, 4, 5)`,表示它有 3 个维度,每个维度分别有 4、5、6 个元素。
阅读全文