import torch
num_classes = 3
std = torch.Tensor([0.1, 0.1, 0.2, 0.2]).repeat(num_classes + 1)[None]
print(std.size())
如果在末尾加一个[None],print出来的结果是:
torch.Size([1, 16])
但是如果没有加[None],
import torch
num_classes = 3
std = torch.Tensor([0.1, 0.1, 0.2, 0.2]).repeat(num_classes + 1)#[None]
print(std.size())
print出来的结果是:
torch.Size([16])
std:
tensor([0.1000, 0.1000, 0.2000, 0.2000, 0.1000, 0.1000, 0.2000, 0.2000, 0.1000,
0.1000, 0.2000, 0.2000, 0.1000, 0.1000, 0.2000, 0.2000])
repeat()操作:将该行向量整体复制后拼接在一起。
[None]操作:降低张量的维度。