We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
参考示例脚本替换为自己得mu和cpp文件,
python导入时报错如下: 包版本如下: setuptools 68.2.2 torch 2.0.0 torch-musa 1.1.0 torchaudio 2.0.1+3b40834 torchvision 0.15.2a0+fa99a53
The text was updated successfully, but these errors were encountered:
请这样使用,
import torch import torch_musa from freqencoder import FreqEncoder view_encoder = FreqEncoder(input_dim=3, degree=4) dirs_musa = torch.tensor([[1.1,2.2,3.3],[4.4,5.5,6.6]], requires_grad=True, device="musa") def compute(dirs): dirs_ = dirs/dirs.norm(dim=-1,keepdim=True) code = view_encoder(dirs_) loss = code.mean() loss.backward() return code print(compute(dirs_musa), dirs_musa.grad)
Sorry, something went wrong.
No branches or pull requests
参考示例脚本替换为自己得mu和cpp文件,
python导入时报错如下:
包版本如下:
setuptools 68.2.2
torch 2.0.0
torch-musa 1.1.0
torchaudio 2.0.1+3b40834
torchvision 0.15.2a0+fa99a53
The text was updated successfully, but these errors were encountered: