摘要: 假设我有一个tensor,它的batch_size是2: tensor = torch.randn([2, 6]) print(tensor.shape) 输出是 torch.Size([2, 6]) 其中tensor.shape[0]代表tensor的batch_size 如果我要把其中每个Bat 阅读全文
posted @ 2022-09-18 10:11 猪猪猪猪侠 阅读(368) 评论(0) 推荐(0) 编辑