a = torch.tensor([1, 2, 3]) b = a.unsqueeze(0) print(b.shape) # 输出 (1, 1, 2, 3)
时间: 2023-10-23 19:14:55 浏览: 102
TensorFlow打印输出tensor的值
这段代码有误,应该是输出 (1, 3),因为在将张量a调用unsqueeze(0)时,会在第0维前面添加一个维度,因此张量a的形状从原来的(3,)变成了(1, 3)。因此,b的形状为(1, 3)。如果想要让输出为(1, 1, 3),需要使用unsqueeze(0)两次,即b = a.unsqueeze(0).unsqueeze(0)。
阅读全文