torch.nn.init torch.nn.init torch . nn . init . calculate_gain ( nonlinearity , param = None ) 对于给定的非线性函数，返回推荐的增益值。这些值如下所示： nonlinearity gain linear 1 conv{1,...

初始化器的用法 可用的初始化器 Initializer Zeros Ones Constant RandomNormal RandomUniform TruncatedNormal VarianceScaling Orthogonal Identity lecun_uniform glorot_normal glorot_unifo...

初始化器的用法 可用的初始化器 Initializer Zeros Ones Constant RandomNormal RandomUniform TruncatedNormal VarianceScaling Orthogonal Identity lecun_uniform glorot_normal glorot_unifo...

torch.Tensor abs() abs_() acos() acos_() add(value) add_(value) addbmm(beta=1, mat, alpha=1, batch1, batch2) addbmm_(beta=1, mat, alpha=1, batch1, batch2) addcdiv(value=1, ...

meshgrid meshgrid paddle. meshgrid ( args, kargs* ) [源代码] 该OP的输入是张量或者包含张量的列表, 包含 k 个一维张量，对每个张量做扩充操作，输出 k 个 k 维张量。 参数： args （Tensor|Tensor数组）- 输入变量为 k 个一维张量，形状分别为(N1,), ...

meshgrid meshgrid paddle.meshgrid ( \args, **kargs* ) [源代码] 该OP的输入是张量或者包含张量的列表, 包含 k 个一维张量，对每个张量做扩充操作，输出 k 个 k 维张量。 参数： * args （Tensor|Tensor数组）- 输入变量为 k 个一维张量，形状分别为(N1,...

CUDA 语义 内存管理 最佳实践 设备无关代码 使用固定的内存缓冲区 使用 nn.DataParallel 替代 multiprocessing CUDA 语义 译者：@Chris 校对者：@Twinkle torch.cuda 被用于设置和运行 CUDA 操作. 它会记录当前选择的 GPU, 并且分配的所有 CUDA ...

torch.sparse 函数 torch.sparse 译者：hijkzzz 警告 这个API目前还处于试验阶段, 可能在不久的将来会发生变化. Torch支持COO(rdinate )格式的稀疏张量, 这可以有效地存储和处理大多数元素为零的张量. 稀疏张量表示为一对稠密张量:一个值张量和一个二维指标张量. 一个稀疏张量可以通...

广播语义 一般语义 直接语义 (In-place semantics) 向后兼容 广播语义 译者：@谢家柯 @Twinkle 一些 PyTorch 的操作支持基于 NumPy Broadcasting Semantics . 简而言之, 如果一个 PyTorch 操作支持广播语义, 那么它的张量参数可以自动扩展为相同的大小 (...