Closed echosprint closed 1 year ago
torch.zeros(..., dtype=int) will create a tensor with dtype torch.int64, we want a tensor of int32, need to specify the datatype torch.int
torch.zeros(..., dtype=int) will create a tensor with dtype torch.int64, we want a tensor of int32, need to specify the datatype torch.int