那抹阳光1994

摘要： torch.eye(n, m=None, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor Returns a 2-D tensor with ones on the d 阅读全文

摘要： Tensor.expand(*sizes) → 张量 返回自张量的新视图，单例维度扩展到更大的尺寸。 传递 -1 作为维度的大小意味着不更改该维度的大小。 Tensor 也可以扩展到更多的维度，新的维度会附加在前面。 对于新维度，大小不能设置为 -1。 扩展张量不会分配新的内存，而只会在现有张量上创 阅读全文