摘要： 作用：收集输入的特定维度指定位置的数值参数：input(tensor): 待操作数。不妨设其维度为（x1, x2, …, xn）dim(int): 待操作的维度。index(LongTensor): 如何对input进行操作。其维度有限定，例如当dim=i时，index的维度为（x1, x2, …y 阅读全文

摘要： a = torch.range(1,4) a = a.reshape(2,1,2) b= torch.range(1,12) b = b.reshape(2,2,3) c = torch.bmm(a,b) print('c') print(c) print(c.shape) d = torch.ze 阅读全文