小编vah*_*ndi的帖子

我想指定类型提示的功能,可以接收无论是list的strS或A list的int秒.有没有办法用Python做到这一点？就像是:

from typing import List

def my_function(list_arg: List[str|int]):

    ...
    ...

我正在PyMC3计算一些我不会在这里讨论的内容，但是如果您有兴趣的话，可以从此链接中了解。

“ 2-lambdas”情况基本上是一个开关函数，需要将其编译为一个Theano函数以避免dtype错误，并且看起来像这样：

import theano
from theano.tensor import lscalar, dscalar, lvector, dvector, argsort

@theano.compile.ops.as_op(itypes=[lscalar, dscalar, dscalar], otypes=[dvector])
def lambda_2_distributions(tau, lambda_1, lambda_2):
        """
        Return values of `lambda_` for each observation based on the 
        transition value `tau`.
        """
        out = zeros(num_observations)
        out[: tau] = lambda_1  # lambda before tau is lambda1
        out[tau:] = lambda_2  # lambda after (and including) tau is lambda2
        return out

我试图将其推广到适用于'n-lambdas'的地方taus.shape[0] = lambdas.shape[0] - 1，但是我只能提出这种极其缓慢的numpy实现。

@theano.compile.ops.as_op(itypes=[lvector, …