摘要: 这里实现了一下常见的激活函数。 其中prelu参数是一个可学习参数,这里设成前向传播类。 代码如下: import numpy as np import matplotlib.pyplot as plt import torch import torch.nn as nn def sigmoid(x 阅读全文
posted @ 2024-08-17 22:05 Dsp Tian 阅读(8) 评论(0) 推荐(0) 编辑
摘要: 写了一个计算距离的脚本,常见距离基本都有。 其中测地距离需要依赖曲面,Hausdorff距离之前有实现,而Wasserstei距离可以用sinkhorn方法求解。 代码如下: import numpy as np def Euclidean(a,b): return np.sqrt(np.sum(( 阅读全文
posted @ 2024-08-17 17:42 Dsp Tian 阅读(3) 评论(0) 推荐(0) 编辑