python语法
- X.sum(维度, keepdim=True or False)
- X.argmax(axis = …)
- isinstance(a, b) a是不是属于b数据类型
- x.numel() 返回数组中元素的个数
- zip(a, b) : a,b都是列表,返回他们对应位置上的元素组成的元组。
- flatten() 返回一个一维函数
- enumerate(list) 返回序号和元素值
- is_tensor() 是tensor返回true
Code Tips
- 矩阵相乘:为了相乘方便,不需要计算维度,只需要X.reshape((-1, X.shape[0]))
- tensor/array 的索引:
X = [[1, 2, 3], [4, 5, 6]]
X[[0, 1], [0, 2]] # 是说索引0,1行中的第0, 第2个元素 - 写一个迭代器 [9_softmax_0