indices should be either on cpu or on the same device as the indexed tensor (cpu)
indices should be either on cpu or on the same device as the indexed tensor (cpu)
WeChat Self-Service
WeChat Consult
TaoBao
support@elephdev.com
Top