torch.Size理解

170 阅读1分钟
  1. torch.Size括号中有几个数字就是几维
  2. 第一层(最外层)中括号里面包含了两个中括号(以逗号进行分割),这就是(2,3,4)中的2
    第二层中括号里面包含了三个中括号(以逗号进行分割),这就是(2,3,4)中的3
    第三层中括号里面包含了四个数(以逗号进行分割),这就是(2,3,4)中的4
    在这里插入图片描述