看深度学习代码遇见的一些torch用法

torch.eq 对两个张量Tensor进行逐元素的比较，若相同位置的两个元素相同，则返回True；若不同，返回False。
torch.matmul 两个张量的乘法。
torch.div 数组的点除运算。
torch.max 返回最大值和最大值的位置索引。
torch.ones_like(mask) 返回一个填充了标量值1的张量，其大小与mask相同。
torch.log是以自然数e为底的对数函数。

logits_mask = torch.scatter(
    torch.ones_like(mask),  # 返回一个填充了标量值1的张量，其大小与mask相同
    1,
    torch.arange(batch_size).view(-1, 1).to(device),  # arrange返回大小为(end-start)/step的一维张量
    0
)

参考链接https://pythontechworld.com/article/detail/7abfpMKf4LPP#:~:text=%23torch.scatter%E5%87%BD%E6%95%B0%E5%AE%98%E6%96%B9%E8%A7%A3%E9%87%8A%20scatter%20%28output%2C%20dim%2C%20index%2C%20src%29%20%E2%86%92%20Tensor,at%20the%20indices%20specified%20in%20the%20index%20tensor.