torch.fmod

引用
#### 引用[.reference_title]
- *1* *2* *3* [Pytorch学习之torch----数学操作(二)](https://blog.csdn.net/gyt15663668337/article/details/91388593)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]

相关问题

torch.floor

torch.floor函数的作用是返回张量中每个元素的向下取整值。它的参数input是输入张量，out是可选的输出张量。例如，如果我们有一个输入张量a=torch.randn(4)，调用torch.floor(a)会返回一个新的张量，其中包含a中每个元素的向下取整值。

另外，torch.fmod函数用于计算输入张量与除数之间的模运算。它的参数input是被除数，divisor可以是一个数或与被除数相同类型的张量，out是可选的输出张量。例如，调用torch.fmod(torch.Tensor([1, 2, 3, 4, 5]), 1.5)会返回一个新的张量，其中包含每个元素与1.5之间的模运算结果。

还有一个相关的函数是torch.frac，它用于计算输入张量中每个元素的小数部分。参数input是输入张量，out是可选的输出张量。例如，如果我们有一个输入张量a=torch.randn(4)，调用torch.frac(a)会返回一个新的张量，其中包含a中每个元素的小数部分。

最后，torch.mul函数用于将输入张量的每个元素乘以另一个张量或一个数。它的参数input是输入张量，other可以是另一个张量或一个数，out是可选的输出张量。例如，调用torch.mul(a, 100)会返回一个新的张量，其中包含a中每个元素乘以100的结果。

if args.b_distribution == 'none': att_b_ = (att_b * 2 - 1) * args.thres_int if args.b_distribution == 'uniform': att_b_ = (att_b * 2 - 1) * \ torch.rand_like(att_b) * \ (2 * args.thres_int) if args.b_distribution == 'truncated_normal': att_b_ = (att_b * 2 - 1) * \ (torch.fmod(torch.randn_like(att_b), 2) + 2) / 4.0 * \ (2 * args.thres_int) if (it+1) % (args.n_d+1) != 0: errD = attgan.trainD(img_a, att_a, att_a_, att_b, att_b_) add_scalar_dict(writer, errD, it+1, 'D') else: errG = attgan.trainG(img_a, att_a, att_a_, att_b, att_b_) add_scalar_dict(writer, errG, it+1, 'G') progressbar.say(epoch=epoch, iter=it+1, d_loss=errD['d_loss'], g_loss=errG['g_loss'])

这段代码包含了两个条件语句块。

第一个条件语句块根据命令行参数 `args.b_distribution` 的值选择不同的属性 B 分布。如果 `args.b_distribution` 的值为 `'none'`，则执行第一个条件分支。在该分支中，通过 `(att_b * 2 - 1) * args.thres_int` 对属性 B 进行归一化处理，将其缩放到 -`args.thres_int` 到 `args.thres_int` 的范围内。

如果 `args.b_distribution` 的值为 `'uniform'`，则执行第二个条件分支。在该分支中，先通过 `(att_b * 2 - 1)` 对属性 B 进行归一化处理，然后使用 `torch.rand_like(att_b)` 生成与属性 B 相同形状的随机张量，并且每个元素的值在 [0, 1) 的范围内。最后，通过 `(2 * args.thres_int)` 将随机张量缩放到 -`args.thres_int` 到 `args.thres_int` 的范围内。

如果 `args.b_distribution` 的值为 `'truncated_normal'`，则执行第三个条件分支。在该分支中，先通过 `(att_b * 2 - 1)` 对属性 B 进行归一化处理，然后使用 `torch.fmod(torch.randn_like(att_b), 2) + 2` 生成一个符合截断正态分布的随机张量，然后除以 4.0 进行归一化处理。最后，通过 `(2 * args.thres_int)` 将随机张量缩放到 -`args.thres_int` 到 `args.thres_int` 的范围内。

接下来，根据 `(it+1) % (args.n_d+1) != 0` 的条件判断选择执行训练判别器或者生成器的代码块。

如果条件为真，则执行训练判别器的代码块。调用 `attgan.trainD()` 对判别器进行训练，传入图像数据 `img_a`、属性数据 `att_a`、处理后的属性数据 `att_a_`、属性数据 `att_b` 和处理后的属性数据 `att_b_`。将得到的误差值 `errD` 通过 `add_scalar_dict()` 添加到摘要信息中，并指定摘要信息的名称为 `'D'`。

如果条件为假，则执行训练生成器的代码块。调用 `attgan.trainG()` 对生成器进行训练，传入相同的参数。将得到的误差值 `errG` 通过 `add_scalar_dict()` 添加到摘要信息中，并指定摘要信息的名称为 `'G'`。然后，通过 `progressbar.say()` 打印出当前 epoch、迭代次数、判别器损失和生成器损失。

总结起来，这段代码根据命令行参数选择不同的属性 B 分布，并对属性数据进行归一化处理。然后，根据迭代次数的奇偶性选择训练判别器或者生成器，并将得到的误差值记录到摘要信息中。最后，打印出当前的训练进度和损失值。