# -*- coding: utf-8 -*-
import torch
# 转置
a = torch.arange(6).reshape(2, 3)
print('a:{}'.format(a))
b = torch.einsum('ij->ji', [a])
print(b.size())
print(b)
# 求和
b <
pytorch的einsum
最新推荐文章于 2024-08-20 21:41:22 发布
# -*- coding: utf-8 -*-
import torch
# 转置
a = torch.arange(6).reshape(2, 3)
print('a:{}'.format(a))
b = torch.einsum('ij->ji', [a])
print(b.size())
print(b)
# 求和
b <