新南威尔士comp9444可以辅导吗?实现三个part函数
part1: simple_addition函数,两个张量分素相加,得到维度一致的张量,结果张量中每个标量的值是相应标量的和 可使用torch.add方法
simple_reshape函数,将给定的张量重塑为给定的形状,返回的张量必须与原张量有相同的数据和相同的元素个数,但是可以有不同的尺寸,使用torch.Size将张量变形
simple_transpose函数,交换矩阵的第一维度与第二维度,可使用torch.permute方法
simple_concatenate函数,返回一个将一维张量的给定序列连接起来的结果,可使用torch.cat方法,在指定的维度dim上对序列seq进行连接操作。
simple_stack函数,实现一个函数,返回将给定张量序列沿着新的维(dim)串联起来的结果,可使用torch.stack方法。
part2:activation函数,sigmod公式可表示为1 / (1 + np.exp(-x))