Hi @ptrblck,
Yes. The batch.reward is an array/tuple of tensors. Some of the tensors have dtype = torch.int32 and some has dtype = torch.int64.batch.reward type974×268 17.1 KB
I just want to have all tensors in batch.rewards as Int. Am i doing this correct as shown below?
count = 0
For single_reward in batch.reward:
{if count == 0: reward_tuple = single_reward.int()
else: reward_tuple = torch.cat([reward_tuple, single_reward.int()], dim=0) #add tensor to tuple
count= count + 1 }
batch.reward = reward_tuple