I found the output of nn.Embedding is default to be ‘float32’
but I need it to be ‘float64’
CLASS torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2.0, scale_grad_by_freq=False, sparse=False, _weight=None)

and torch documents don’t provide the way to set attr as ‘dtype=’
how can I change that?