Skip to content

torch multinomial returns floats on GPU #261

Closed
@glample

Description

@glample

torch.multinomial(torch.FloatTensor(10).cuda().normal_(), 3)is returning a FloatTensor on GPU (as opposed to a LongTensor on CPU)

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions