Awesome, thank you so much for the quick response! A question related to size then:
torch.unsqueeze(tensor, 0) adds a 1 dimensional object to the beginning of the tensor?
So tensor_test = (2, 2, 2)
tensor_test.unsqueeze(0) then makes tensor_test = (1, 2, 2, 2)?