2022年1月10日
torch.nn.Identity()
摘要： 占位层 阅读全文
posted @ 2022-01-10 14:16 小艾衰 阅读(212) 评论(0) 推荐(0) 编辑
Torch 运算符 @ 和 *
摘要： '@' ： 矩阵乘 '*' ： 矩阵点乘 >>a = torch.tensor([[1, 2], [3, 4]]) >>b = torch.tensor([[4, 5], [6, 7]]) >>print(f'a x b = {a @ b}\na * b = {a * b}') >>a x b = 阅读全文
posted @ 2022-01-10 11:17 小艾衰 阅读(1249) 评论(0) 推荐(0) 编辑