2023年9月26日
如何判断torch中tensor的维度
摘要： 省流版本 在输出一个张量时，你可以根据张量输出的方括号（[]）的数量来判断张量的维度。每个方括号对应张量的一个维度，张量的维度数就是方括号的数量。 最外面的为第一个维度，往里面依次为第2 3 ... 维度。 例如，如果你看到如下输出： tensor([[1, 2, 3], [4, 5, 6]]) 这 阅读全文
posted @ 2023-09-26 11:17 诗子黎 阅读(222) 评论(0) 推荐(0) 编辑