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