初学炼丹，在 Python 里可以这样设置随机数 seed 吗？

[zxCoder]

def set_seed(seed):
    if seed:
        import torch
        torch.random.manual_seed(seed)

main.py 里调用了另外一个 py 文件的这个函数，这样写是对的吗？设置的这个 seed 没有作用域啥的吗