32 lines
561 B
Python
32 lines
561 B
Python
|
import torch
|
||
|
|
||
|
x = torch.tensor([[1, 2], [3, 4]])
|
||
|
|
||
|
print(x)
|
||
|
print("x.tile((2)) -> ", x.tile((2)).shape)
|
||
|
print(x.tile((2)))
|
||
|
|
||
|
print()
|
||
|
print("x.tile((2, 1)) -> ", x.tile((2, 1)).shape)
|
||
|
print(x.tile((2, 1)))
|
||
|
|
||
|
print()
|
||
|
print("x.tile((2, 1, 2)) -> ", x.tile((2, 1, 2)).shape)
|
||
|
print(x.tile((2, 1, 2)))
|
||
|
|
||
|
print()
|
||
|
print("x.tile((2, 1, 1)) -> ", x.tile((2, 1, 1)).shape)
|
||
|
print(x.tile((2, 1, 1)))
|
||
|
|
||
|
print()
|
||
|
y = torch.tensor([[2, 1], [3, 4]])
|
||
|
print(y.ne(x))
|
||
|
|
||
|
print()
|
||
|
print(x.prod(1))
|
||
|
print(x.prod(0))
|
||
|
|
||
|
print()
|
||
|
print(x.unsqueeze(1).shape)
|
||
|
print(x.unsqueeze(1).squeeze(1).shape)
|