I am trying to recreate the following array in pure torch. How can I achieve it using nothing but PyTorch?
t_list= float('inf') * np.ones(np.append( 10, 3+ 1 + np.arange( 2)))
I am trying to recreate the following array in pure torch. How can I achieve it using nothing but PyTorch?
t_list= float('inf') * np.ones(np.append( 10, 3+ 1 + np.arange( 2)))
@Harry-Garrison I hope this works for you.
t_list = float('inf') * torch.ones(torch.cat( [torch.tensor([10]), 3+1+torch.arange( 2)] ).tolist())
Excellent, thank you!