Split einen Tensor in Taschenlampe

Question

Gegeben ein Eingang Tensor der Größe n x 2A x B x C, wie man es in zwei Tensoren teilen, jede von Größe n x A x B x C? Im Wesentlichen ist n die Losgröße.

Answer 1

Ich glaube, Sie tun könnte so etwas wie:

tensor_a = torch.Tensor(n, 2A, B,C) 
-- Initialize tensor_a with the data 

tensor_b = torch.Tensor(n, A, B, C) 
tensor_b = tensor_a[{{},1,{},{}}] 
tensor_c = torch.Tensor(n, A, B, C) 
tensor_c = tensor_a[{{},2,{},{}}]