all in PyTorch

Buy Me a Coffee

*My post explains any().

all() can check if all the elements of a 0D or more D tensor are True, getting the 0D or more D tensor of zero or more elements as shown below:

*Memos:

• all() can be used with torch or a tensor.
• The 1st argument(input) with torch or using a tensor(Required-Type:tensor of int, float, complex or bool).
• The 2nd argument with torch or the 1st argument with a tensor is dim(Optional-Type:int, tuple of int or list of int).
• The 3rd argument with torch or the 2nd argument with a tensor is keepdim(Optional-Default:False-Type:bool). *My post explains keepdim argument.
• There is out argument with torch(Optional-Default:None-Type:tensor): *Memos:
  • out= must be used.
  • My post explains out argument.
• An empty tensor returns a True of a 1D or more D tensor or an empty 1D or more D tensor.

<span>import</span> <span>torch</span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>(</span><span>True</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>my_tensor</span><span>.</span><span>all</span><span>()</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor(True) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>],</span>
                          <span>[</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,</span> <span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,</span> <span>-</span><span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>1</span><span>,</span> <span>0</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>1</span><span>,</span> <span>-</span><span>2</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,</span> <span>0</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,</span> <span>-</span><span>2</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>2</span><span>,</span> <span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>2</span><span>,</span> <span>-</span><span>1</span><span>))</span>
<span># tensor(False) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>2</span><span>)</span>
<span># tensor([True, False, True, False]) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor([False, False]) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0</span><span>,</span> <span>1</span><span>,</span> <span>2</span><span>,</span> <span>3</span><span>],</span>
                          <span>[</span><span>4</span><span>,</span> <span>5</span><span>,</span> <span>6</span><span>,</span> <span>7</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0.</span><span>,</span> <span>1.</span><span>,</span> <span>2.</span><span>,</span> <span>3.</span><span>],</span>
                          <span>[</span><span>4.</span><span>,</span> <span>5.</span><span>,</span> <span>6.</span><span>,</span> <span>7.</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0.</span><span>+</span><span>0.j</span><span>,</span> <span>1.</span><span>+</span><span>0.j</span><span>,</span> <span>2.</span><span>+</span><span>0.j</span><span>,</span> <span>3.</span><span>+</span><span>0.j</span><span>],</span>
                          <span>[</span><span>4.</span><span>+</span><span>0.j</span><span>,</span> <span>5.</span><span>+</span><span>0.j</span><span>,</span> <span>6.</span><span>+</span><span>0.j</span><span>,</span> <span>7.</span><span>+</span><span>0.j</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(True) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>2</span><span>)</span>
<span># tensor([], dtype=torch.bool) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span># tensor([True]) </span>
<span>import</span> <span>torch</span>

<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>(</span><span>True</span><span>)</span>

<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>my_tensor</span><span>.</span><span>all</span><span>()</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor(True) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>])</span>

<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>],</span>
                          <span>[</span><span>True</span><span>,</span> <span>False</span><span>,</span> <span>True</span><span>,</span> <span>False</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,</span> <span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,</span> <span>-</span><span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>1</span><span>,</span> <span>0</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>1</span><span>,</span> <span>-</span><span>2</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,</span> <span>0</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,</span> <span>-</span><span>2</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>2</span><span>,</span> <span>1</span><span>))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>2</span><span>,</span> <span>-</span><span>1</span><span>))</span>
<span># tensor(False) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>0</span><span>,))</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>2</span><span>)</span>
<span># tensor([True, False, True, False]) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>(</span><span>-</span><span>1</span><span>,))</span>
<span># tensor([False, False]) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0</span><span>,</span> <span>1</span><span>,</span> <span>2</span><span>,</span> <span>3</span><span>],</span>
                          <span>[</span><span>4</span><span>,</span> <span>5</span><span>,</span> <span>6</span><span>,</span> <span>7</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0.</span><span>,</span> <span>1.</span><span>,</span> <span>2.</span><span>,</span> <span>3.</span><span>],</span>
                          <span>[</span><span>4.</span><span>,</span> <span>5.</span><span>,</span> <span>6.</span><span>,</span> <span>7.</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[</span><span>0.</span><span>+</span><span>0.j</span><span>,</span> <span>1.</span><span>+</span><span>0.j</span><span>,</span> <span>2.</span><span>+</span><span>0.j</span><span>,</span> <span>3.</span><span>+</span><span>0.j</span><span>],</span>
                          <span>[</span><span>4.</span><span>+</span><span>0.j</span><span>,</span> <span>5.</span><span>+</span><span>0.j</span><span>,</span> <span>6.</span><span>+</span><span>0.j</span><span>,</span> <span>7.</span><span>+</span><span>0.j</span><span>]])</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(False) </span>
<span>my_tensor</span> <span>=</span> <span>torch</span><span>.</span><span>tensor</span><span>([[]])</span>

<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>)</span>
<span># tensor(True) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>0</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>2</span><span>)</span>
<span># tensor([], dtype=torch.bool) </span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=</span><span>1</span><span>)</span>
<span>torch</span><span>.</span><span>all</span><span>(</span><span>input</span><span>=</span><span>my_tensor</span><span>,</span> <span>dim</span><span>=-</span><span>1</span><span>)</span>
<span># tensor([True]) </span>
import torch

my_tensor = torch.tensor(True)

torch.all(input=my_tensor)
my_tensor.all()
torch.all(input=my_tensor, dim=0)
torch.all(input=my_tensor, dim=-1)
torch.all(input=my_tensor, dim=(0,))
torch.all(input=my_tensor, dim=(-1,))
# tensor(True) 
my_tensor = torch.tensor([True, False, True, False])

torch.all(input=my_tensor)
torch.all(input=my_tensor, dim=0)
torch.all(input=my_tensor, dim=-1)
torch.all(input=my_tensor, dim=(0,))
torch.all(input=my_tensor, dim=(-1,))
# tensor(False) 
my_tensor = torch.tensor([[True, False, True, False],
                          [True, False, True, False]])
torch.all(input=my_tensor)
torch.all(input=my_tensor, dim=(0, 1))
torch.all(input=my_tensor, dim=(0, -1))
torch.all(input=my_tensor, dim=(1, 0))
torch.all(input=my_tensor, dim=(1, -2))
torch.all(input=my_tensor, dim=(-1, 0))
torch.all(input=my_tensor, dim=(-1, -2))
torch.all(input=my_tensor, dim=(-2, 1))
torch.all(input=my_tensor, dim=(-2, -1))
# tensor(False) 
torch.all(input=my_tensor, dim=0)
torch.all(input=my_tensor, dim=(0,))
torch.all(input=my_tensor, dim=-2)
# tensor([True, False, True, False]) 
torch.all(input=my_tensor, dim=1)
torch.all(input=my_tensor, dim=-1)
torch.all(input=my_tensor, dim=(-1,))
# tensor([False, False]) 
my_tensor = torch.tensor([[0, 1, 2, 3],
                          [4, 5, 6, 7]])
torch.all(input=my_tensor)
# tensor(False) 
my_tensor = torch.tensor([[0., 1., 2., 3.],
                          [4., 5., 6., 7.]])
torch.all(input=my_tensor)
# tensor(False) 
my_tensor = torch.tensor([[0.+0.j, 1.+0.j, 2.+0.j, 3.+0.j],
                          [4.+0.j, 5.+0.j, 6.+0.j, 7.+0.j]])
torch.all(input=my_tensor)
# tensor(False) 
my_tensor = torch.tensor([[]])

torch.all(input=my_tensor)
# tensor(True) 
torch.all(input=my_tensor, dim=0)
torch.all(input=my_tensor, dim=-2)
# tensor([], dtype=torch.bool) 
torch.all(input=my_tensor, dim=1)
torch.all(input=my_tensor, dim=-1)
# tensor([True]) 

Enter fullscreen mode Exit fullscreen mode

原文链接：all in PyTorch