关于tensor维度的实验

dataset.get_styles()  #{Tensor:(100,64)}
all_style_inds        #{Tensor:(4800,)}
dataset.get_styles()[all_style_inds]  #{Tensor:(4800,64)}

可以发现[]相当于在dataset.get_styles()第一个维度上对每一个all_style_inds的数值。