Hi, all,
when I tried to implement:
m = nn.Hardshrink()
input = Variable(torch.randn(2)).cuda()
print(m(input))


It showed hard shrink_forward is not implemented for type torch.cuda.FloatTensor.
However, when I tried Softshrink, it worked. Does this mean this function doesn’t support cuda?
Thanks in advance!