解包操作
可变参数和关键字参数是 Python 函数的两种参数类型。
b, *_ = t.shape python中这个用法是什么意思
我们使用 _” 来“忽略”一个或多个值(表示我们不需要这些值),然后将“t.shape”元组的第一个元素赋值给变量“b`”。
pytorcg gather函数
可参考这篇文章https://www.jianshu.com/p/2ff9a6905abe
torch.cat()函数
*shape[1:]前面的星号是什么意思
星号 * 是一个解包运算符,它可以将一个可迭代对象拆分成单个元素。在给定的上下文中,shape 是一个元组,shape[1:] 表示从索引1开始的所有元素,即排除第一个元素。在这种情况下使用星号 * 可以将这些元素解包为单独的参数,以便将它们传递给一个接受单独参数的函数。例如,假设 shape 是 (3, 4, 5) ,那么 shape[1:] 将得到 (4, 5),而 *shape[1:] 将解包为 4, 5。
torch.randn_like()解释一下
torch.randint()
numpy.cumprod(a, axis=None, dtype=None, out=None)
可参考这篇文章https://vimsky.com/examples/usage/python-numpy.cumprod.html