import torch
torch.manual_seed(2)
x = torch.randn(1, 2, 3,4)
A = torch.randn(3, 2)
torch.einsum()用法举例
最新推荐文章于 2024-03-20 13:10:53 发布
import torch
torch.manual_seed(2)
x = torch.randn(1, 2, 3,4)
A = torch.randn(3, 2)