@TOC
代码
import numpy as np
import torch
a = torch.from_numpy(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
b = torch.from_numpy(np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]]))
c = torch.from_numpy(np.array([[11, 21, 31], [41, 51, 61], [71, 81, 91]]